python - Pandas Dataframe For 循环

Question

我想用另一列的一部分替换 pandas 数据框中的一列。到目前为止我所拥有的是:

for index, row in df.iterrows():
  File = df.at[row, 'FileName']
  df.at[row, 'NUMBER'] = File.split(".")[1]

理想情况下，这将遍历数据框的行并将数字列替换为文件名列的一部分

我收到错误:

ValueError: At based indexing on an integer index can only have integer indexers

我认为这与 df.at[] 的误用有关，但我不确定如何修复它。

Answer 1

不要按iterrows 循环，因为slow , 更好的是使用 str.split通过索引选择第二个列表:

df['NUMBER'] = df['FileName'].str.split(".").str[1]

如果需要更好的性能，或者使用list comprehension:

df['NUMBER'] = [x.split(".")[1] for x in df['FileName']]