例如,我有一个包含名称和颜色的水果数据集。如何根据选择的水果名称更改颜色列中的值?
Name Color
Apple NaN
Pear Green
Pear Green
Banana Yellow
Watermelon Green
我有一个大概的主意,但我不知道如何编码。
df[Name] == Apple then df[color] == Red
一种方法是使用.apply():
In [83]: df
Out[83]:
Name Color
0 Apple NaN
1 Pear Green
2 Pear Green
3 Banana Yellow
4 Watermelon Green
In [84]: df['Color'] = df.apply(lambda x: 'Red' if x.Name == 'Apple' else x.Color, 1)
In [85]: df
Out[85]:
Name Color
0 Apple Red
1 Pear Green
2 Pear Green
3 Banana Yellow
4 Watermelon Green
另一个更简单的方法是使用.ix:
In [94]: df
Out[94]:
Name Color
0 Apple NaN
1 Pear Green
2 Pear Green
3 Banana Yellow
4 Watermelon Green
In [95]: df.ix[df.Name == 'Apple', 'Color'] = 'Red'
In [96]: df
Out[96]:
Name Color
0 Apple Red
1 Pear Green
2 Pear Green
3 Banana Yellow
4 Watermelon Green
本文收集自互联网,转载请注明来源。
如有侵权,请联系 [email protected] 删除。
我来说两句