將 Pandas DataFrame 列值從數組更改為數組中的第一個值

Question

我有一個帶有labels列的 DataFrame。 labels列當前是一個數組，我想將整列 map 到每一行的數組的第一項，而不是天真地迭代。

例如，如果我有 2 行帶有labels值 ['1'，'m']，['0']，我想將當前值 map 轉換為新值 '1'，'0'。

Answer 1

您可以為此使用.str ：

df['labels'] = df['labels'].str[0]

然而，它（和其他可能的方式）本質上只是一個循環：

df['labels'] = [x[0] for x in df['labels'] ]

我建議做一個循環，因為您可以更好地控制錯誤處理，例如：

# this handles empty array as well as NaN values
df['labels'] = [x[0] if x else np.nan for x in df['labels']]