通過部分字符串在 pandas dataframe 列表中查找字符串

Question

我需要根據元素的一部分在每個列列表中查找元素。 a1應該找到a值。 如果column1是單個字符串並且如果column1中的列表不起作用，則contains工作

原始數據框：

期望結果：

這是我嘗試過的代碼：

frame = pd.DataFrame({'column1' : [['a-1','b-1','c-1'], ['a-2','b-2','c-2'], ['a-3','b-3','c-3']]})
frame['column1']=frame[frame['column1'].str.contains('a')]

Answer 1

如果列表的順序沒有改變，你可以嘗試類似

frame['column1'] = frame['column1'].str[0]

frame

Output

   column1
0   a-1
1   a-2
2   a-3