python - python pandas-将值输入到新列

我有一个很小的数据框，无法容纳4人。
有一个称为“成绩”的空列。
我想对那些花费超过100美元的A级和不超过100美元的B级的人进行评分。
假设列“等级”很大，最有效的填充方法是什么？

import pandas as pd
df=pd.DataFrame({'Customer':['Bob','Ken','Steve','Joe'],
             'Spending':[130,22,313,46]})
df['Grade']=''

最佳答案

您可以使用numpy.where：

df['Grade']= np.where(df['Spending'] > 100 ,'A','B')
print (df)
  Customer  Spending Grade
0      Bob       130     A
1      Ken        22     B
2    Steve       313     A
3      Joe        46     B

时间：

df=pd.DataFrame({'Customer':['Bob','Ken','Steve','Joe'],
             'Spending':[130,22,313,46]})

#[400000 rows x 4 columns]
df = pd.concat([df]*100000).reset_index(drop=True)

In [129]: %timeit df['Grade']= np.where(df['Spending'] > 100 ,'A','B')
10 loops, best of 3: 21.6 ms per loop

In [130]: %timeit df['grade'] = df.apply(lambda row: 'A' if row['Spending'] > 100 else 'B', axis = 1)
1 loop, best of 3: 7.08 s per loop

关于python - python pandas-将值输入到新列，我们在Stack Overflow上找到一个类似的问题：https://stackoverflow.com/questions/41165818/