python - 获取应用函数 Pandas 中的第一个列表元素

Question

我有一个preprocess_names拆分输入变量的函数：

def preprocess_names(name):
    #SOME PREPROCESSING
    return name.split('')

另外，我有一个df要应用此功能的 Pandas DataFrame。就我而言，preprocess_names的输出长度为 3，我想在df.

因此，对于第一个新列，我想使用输出列表中的第一个元素。如果我只在一个 str 上使用这个函数，我会用它preprocess_names[0]来获取正确的变量。在将方法应用于 df 时如何做类似的事情？

下面的代码 raises ValueError，但最接近我想做的事情。

df['runs'] = df['name'].apply(preprocess_names)[0]

标签： pythonpandasdataframeapply

Answer 1

你想要：

df['runs'] = df['name'].apply(preprocess_names).str[0]