首页
学习
活动
专区
圈层
工具
发布
首页
学习
活动
专区
圈层
工具
MCP广场
社区首页 >问答首页 >用虚线替换图的一部分

用虚线替换图的一部分
EN

Stack Overflow用户
提问于 2018-01-15 17:21:44
回答 3查看 2.4K关注 0票数 4

我想将函数下降到'-1‘的部分曲线图替换为从上一点开始的虚线(参见下面的曲线图)。

下面是我写的一些代码,以及它的输出:

代码语言:javascript
复制
import numpy as np

import matplotlib.pyplot as plt
y = [5,6,8,3,5,7,3,6,-1,3,8,5]

plt.plot(np.linspace(1,12,12),y,'r-o')
plt.show()

for i in range(1,len(y)):
    if y[i]!=-1:
        plt.plot(np.linspace(i-1,i,2),y[i-1:i+1],'r-o')
    else:
        y[i]=y[i-1]
        plt.plot(np.linspace(i-1,i,2),y[i-1:i+1],'r--o')
plt.ylim(-1,9)
plt.show()

这是原始的情节

修改后的图:

我写的代码可以工作(它会产生所需的输出),但是当我在我的(更大的)数据集上实际运行它时,它的效率很低,并且需要很长时间。有没有更聪明的方法来做这件事呢?

EN

Stack Overflow用户

发布于 2018-01-16 02:51:39

您可以在不使用循环的情况下实现类似的结果:

代码语言:javascript
复制
import pandas as pd
import matplotlib.pyplot as plt

# Create a data frame from the list
a = pd.DataFrame([5,6,-1,-1, 8,3,5,7,3,6,-1,3,8,5])

# Prepare a boolean mask
mask = a > 0

# New data frame with missing values filled with the last element of   
# the previous segment. Choose 'bfill' to use the first element of 
# the next segment.
a_masked = a[mask].fillna(method = 'ffill')

# Prepare the plot
fig, ax = plt.subplots()
line, = ax.plot(a_masked, ls = '--', lw = 1)
ax.plot(a[mask], color=line.get_color(), lw=1.5, marker = 'o')
plt.show()

您还可以通过为线条选择不同的颜色来突出显示负片区域:

我的答案是基于2017年7月的一篇很棒的帖子。后者还处理第一个元素为NaN或在本例中为负数时的情况:Dotted lines instead of a missing value in matplotlib

票数 3
EN
查看全部 3 条回答
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/48260014

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档