我有一个沿着下面的数据框:
Type Set
1 A Z
2 B Z
3 B X
4 C Y
Run Code Online (Sandbox Code Playgroud)
我想在数据帧中添加另一列(或生成一系列)与数据帧相同的长度(=相等的记录/行数),如果Set ='Z'则设置颜色为绿色,如果Set =否则设置为'red' .
最好的方法是什么?
如何检查条件下的所有值是否都col1满足例如> 2?
import pandas as pd
d = [
{'col1': 3, 'col2': 'wasteful'},
{'col1': 0, 'col2': 'hardly'},
]
df = pd.DataFrame(d)
Run Code Online (Sandbox Code Playgroud)
我可以去
if all(col1 > 2 for i, col1, col2 in df.itertuples()):
#do stuff
Run Code Online (Sandbox Code Playgroud)
但是是否有更易读,更快速和/或更少的内存占用方式?