python - 基于 x 轴上给出的新间隔的累积总和的阶梯图
问题描述
我有两个列表: 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....]]
如何修改我定义的函数来做到这一点?
解决方案
我可能会误解问题或期望的结果。我认为您正在寻找的是:
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()
推荐阅读
- java - Jira SDK:atlas-create-jira-plugin 由于 Google Analytics 而失败
- python - 如何获得一个单词的 NER(命名实体识别)
- c++ - 将 main() 更改为 WinMain() 以绘制简单窗口后出现大量错误
- reactjs - Electron 和 React 应用程序 UI 在构建中变得空白
- sql - 如何删除多行 varbinary 数据的第一个字节?
- azure-devops - Azure DevOps C# SDK - 如何更新构建定义的 PR 触发器?
- c# - Mvvmcross xamarin 在 Listview 模板中形成按钮命令
- service - 队列容量为零的服务块
- python - 使用 torch.det 查找行列式不返回 0?
- cocoapods - 您的 Podfile 需要安装插件 `cocoapods-patch`