如何使用熊貓從列中提取類

Question

在最后一列中，我試圖提取並僅顯示一個類。 例如，“卡車”或“汽車”。 有誰知道如何列出它們？

Answer 1

您可以檢查str.contains是否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)]]