如何使用熊猫从列中提取类

Question

In the last column, I am trying to extract and only display a class. For example, 'truck' or 'car'. Do anyone know how to list them?

Answer 1

您可以仅检查str.contains的搜索值

df[df['col_name'].str.contains('truck')]

或者使用链接的str.get并获取出现的内容

df.col1.str.get(0).str.get(0)

示例：

df = pd.DataFrame()
df['col1'] = [[['truck',3, ('a',2)]], [['car', 2, ('b', 2)]]]

    col1
0   [[truck, 3, (a, 2)]]
1   [[car, 2, (b, 2)]]

其中

df.col1.str.get(0).str.get(0)

收益

0    truck
1      car
Name: col1, dtype: object

因此您可以使用loc

df.loc[df.col1.str.get(0).str.get(0).eq('truck')]

    col1
0   [[truck, 3, (a, 2)]]