如果其他列值相等,Pandas会返回列值

时间:2016-09-27 17:28:50

标签: python csv pandas

所以,我试图使用python和pandas从CSV文件中获取数据。 如果第2列的数据等于某个字符串,则代码应从colum 1返回或打印数据。

    import pandas as pd
df = pd.read_csv('alerts.csv', sep=';', encoding='latin1')

print(df[['color']['item']].loc[['color']=='red'])

但它似乎不适用于字符串?

1 个答案:

答案 0 :(得分:0)

你没有正确使用.loc

.loc需要一个索引器,而这样的列有

indexer = df[df['color']=='red'].index
print(df.loc[indexer,'item'])