为数据框中的每个非零值绘制一条垂直线

Question

我需要为数据框中的每个非零值制作垂直线。我使用以下方法创建了一个小数据框：

df = pd.DataFrame(np.random.randint(0,2,size=(100,1)))

我尝试在每个有 1 而不是 0 的地方创建一个带有垂直线的图：

graph = plt.figure(1)
plt.axvline(x = df)
graph.show()

但是，我收到此错误： “ValueError：DataFrame的真值不明确。使用a.empty、a.bool()、a.item()、a.any()或a.all()。”

有人知道怎么做吗？

Answer 1

使用axvline：

import numpy as np
import pandas as pd
import matplotlib.pyplot as plt


df = pd.DataFrame(np.random.randint(0,2,size=(100,1)))

fig, ax = plt.subplots()

for i, row in df.iterrows():
    if row.values != 0:
        ax.axvline(x = i)

plt.show()

使用vlines：

import numpy as np
import pandas as pd
import matplotlib.pyplot as plt


df = pd.DataFrame(np.random.randint(0,2,size=(100,1)))

fig, ax = plt.subplots()

x = df.index.values[(df.values[:,0] > 0)]
ax.vlines(x, 0,1, transform=ax.get_xaxis_transform() )

plt.show()