按字符串的一部分在熊猫数据框列表中查找字符串

Question

我需要根据元素的一部分在每个列列表中找到元素。 a1 应按 a 值查找。 contains 在 column1 是单个字符串的情况下有效，如果 column1 中的列表无效

原始数据帧：

想要的结果：

这是我试过的代码：

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

输出

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