如何使用pandas从列中提取类

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)]]