扩大 pandas 数据框中值之间的距离

Question

如何扩大 pandas 数据框中值之间的距离？

  A
1 3
2 5
3 6
5 5
6 9

我想将相邻元素之间的距离增加x倍，例如两倍。

预期输出：

  A B
1 3 3
2 5 7  # (3 + 2 * 2)
3 6 9  # (7 + 1 * 2)
5 5 7  # (9 - 1 * 2)
6 9 15 # (7 + 4 * 2)

Answer 1

IIUC，你可以尝试：

df['B'] = df['A'].add(df['A'].diff().cumsum()).fillna(df['A'])
print(df)

# Output
   A     B
1  3   3.0
2  5   7.0
3  6   9.0
5  5   7.0
6  9  15.0