根据行值将一行插入熊猫数据框中？答案

【问题标题】：Inserting a row into a pandas dataframe based on row value?根据行值将一行插入熊猫数据框中？
【发布时间】：2016-08-31 17:07:10
【问题描述】：

我有一个数据框：

df = pd.DataFrame({'B':[2,1,2],'C':['a','b','a']})
  B C
0 2 'a'
1 1 'b'
2 2 'a'

我想在任何出现的 'b' 上方插入一行，这是该行的副本，但将 'b' 更改为 'c'，所以我最终得到以下结果：

  B C
0 2 'a'
1 1 'b'
1 1 'c'
2 2 'a'

对于我的一生，我无法弄清楚如何做到这一点。

【问题讨论】：

您在上面说过，但在下面的输出中，您的第一个 df 在第三行生成 a 而不是 c。
如果b连续两行怎么办？
@shivsn 对不起，错字
@Divakar，这不太可能发生，但如果发生了，那么我只需在每个“b”行上方插入一个“c”行
你对循环满意吗？

标签： python pandas numpy dataframe insert

【解决方案1】：

这是一种方法：

duplicates = df[df['C'] == 'b'].copy()
duplicates['C'] = 'c'
df.append(duplicates).sort_index()

【讨论】：

漂亮，没想到还有复制功能！

【解决方案2】：

在 NumPy 级别工作，这是一种矢量化方法 -

arr = df.values
idx = np.flatnonzero(df.C=='b')
newvals = arr[idx]
newvals[:,df.columns.get_loc("C")] = 'c'
out = np.insert(arr,idx+1,newvals,axis=0)
df_index = np.insert(np.arange(arr.shape[0]),idx+1,idx,axis=0)
df_out = pd.DataFrame(out,index=df_index)

示例运行 -

In [149]: df
Out[149]: 
   B  C
0  2  a
1  1  b
2  2  d
3  4  d
4  3  b
5  8  a
6  4  a
7  2  b

In [150]: df_out
Out[150]: 
   0  1
0  2  a
1  1  b
1  1  c
2  2  d
3  4  d
4  3  b
4  3  c
5  8  a
6  4  a
7  2  b
7  2  c

【讨论】：