基于x轴上给出的新间隔的累积总和的阶梯图

Question

我有两个列表：data和Given_x_axis

data=[[0.05, 3200], [0.1, 2000], [0.12, 1200], [0.13, 2000], [0.21, 1800], [0.25, 2800], [0.27, 1500]]
given_x_axis=[0.05, 0.07, 0.09, 0.11, 0.13, 0.15, 0.17, 0.19, 0.21, 0.23, 0.25, 0.27, 0.29, 0.31, 0.33, 0.35]

我要绘制一个具有这样的累计和的步骤图，

x,y=map(list, zip(*np.cumsum(data, axis=0)))
plt.step(x,y)

但使用Given_x_axis作为x轴上的步骤

我试图定义一个函数，该函数根据Given_x_axis重新创建一个新的累积值列表

def update_x_axis(data, given_x_axis):
 cumulated_values=[]
 value_each_step=0
 for n,x in enumerate(given_x_axis):
  for d in data:
   if d[0]<=x:
    value_each_step=value_each_step+d[1]
  cumulated_values.append(value_each_step)
 return [given_x_axis,cumulated_values]

但是y轴上新的累积值列表似乎不正确。 我希望update_x_axis（data，Given_x_axis）会返回

[0.05, 0.07, 0.09, 0.11, 0.13, 0.15, 0.17, 0.19, 0.21, 0.23, 0.25, 0.27, 0.29, 0.31, 0.33, 0.35], 
[3200, 3200, 3200, 5200, 6400, 8400....]]

如何修改定义的函数来做到这一点？

Answer 1

我可能会误解这个问题或期望的结果。我认为您正在寻找的是这样的：

import numpy as np
import matplotlib.pyplot as plt

data=[[0.05, 3200], [0.1, 2000], [0.12, 1200], [0.13, 2000], [0.21, 1800], [0.25, 2800], [0.27, 1500]]
given_x_axis=[0.05, 0.07, 0.09, 0.11, 0.13, 0.15, 0.17, 0.19, 0.21, 0.23, 0.25, 0.27, 0.29, 0.31, 0.33, 0.35]

x,y = np.array(data).T
ind = np.searchsorted(x, given_x_axis, side="left")
ind[ind == 0] = 1
res = np.cumsum(y)[ind-1]

现在是

res

[ 3200.  3200.  3200.  5200.  6400.  8400.  8400.  8400.  8400. 10200.
 10200. 13000. 14500. 14500. 14500. 14500.]

然后绘图

fig, ax = plt.subplots()
ax.plot(x,np.cumsum(y), marker="o", ls="")
ax.step(given_x_axis, res)

plt.show()