从datafram中提取数据

2024-04-26 00:20:50 发布

您现在位置:Python中文网/ 问答频道 /正文

我有一个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().

谁能告诉我我做错了什么吗?非常感谢。你知道吗


Tags: or数据id类型foriftypecolumn
1条回答
网友
1楼 · 发布于 2024-04-26 00:20:50

试试这个: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

相关问题 更多 >