8

使用 DataFrame 的子集进行迭代的最佳方法是什么？

让我们举一个简单的例子：

import pandas as pd

df = pd.DataFrame({
  'Product': list('AAAABBAA'),
  'Quantity': [5,2,5,10,1,5,2,3],
  'Start' : [
      DT.datetime(2013,1,1,9,0),
      DT.datetime(2013,1,1,8,5),
      DT.datetime(2013,2,5,14,0),
      DT.datetime(2013,2,5,16,0),
      DT.datetime(2013,2,8,20,0),                                      
      DT.datetime(2013,2,8,16,50),
      DT.datetime(2013,2,8,7,0),
      DT.datetime(2013,7,4,8,0)]})

df = df.set_index(['Start'])

现在我想使用 itterrows 函数修改这个 DataFrame 的一个子集，例如：

for i, row_i in df[df.Product == 'A'].iterrows():
    row_i['Product'] = 'A1' # actually a more complex calculation

但是，更改不会持续存在。

是否有任何可能性（使用索引“i”进行手动查找除外）对原始 Dataframe 进行持久更改？

4

2 回答 2

2

为什么你需要 iterrows() 呢？我认为在 pandas（或 numpy）中使用矢量化操作总是更可取的：

df.ix[df['Product'] == 'A', "Product"] = 'A1'
于 2013-10-29T18:25:53.470 回答
0

我想我想到的最好方法是生成一个具有所需结果的新向量，您可以在其中循环所有想要的内容，然后将其重新分配回列

#make a copy of the column
P = df.Product.copy()
#do the operation or loop if you really must
P[ P=="A" ] = "A1"
#reassign to original df
df["Product"] = P
于 2017-11-19T15:46:10.797 回答