从数据框中提取数据

时间:2019-07-19 13:14:59

标签: python pandas dataframe

我有一个python数据框,我正试图从中提取一些数据:

frame   id        type         truncated
0       -1      DontCare        -1
0       10        Car            0  
0       13       Misc            0
0       11        Car            1
0       12        Car            1

,我想提取与Car类型有关的数据。所以我要做的是:

 for column in labels['type'].items():
        if column == 'DontCare':
            continue
        if column == "Car" or "Van":
            print('car')
        else:
            print('no car')

但是我得到这个错误:

ValueError: The truth value of a Series is ambiguous. Use a.empty, a.bool(), a.item(), a.any() or a.all().

谁能告诉我我做错了什么?谢谢。

1 个答案:

答案 0 :(得分:0)

尝试一下:df[(df.Type=="Car") | (df.Type=="Van")]

例如,

data = [['Car', 10], ['Van', 15], ['Car', 14], ['DNC', 11]] 
df = pd.DataFrame(data, columns = ['Type', 'Value']) 

print(df) 

产生

    Type    Value
0   Car     10
1   Van     15
2   Car     14
3   DNC     11

print(df[(df.Type=="Car") | (df.Type=="Van")])

产生

    Type    Value
0   Car     10
1   Van     15
2   Car     14