Matplotlib eventplot绘制二元值堆叠栅格图的报错与结果不符问题求助

• plt.eventplot的输入逻辑和预期不匹配：该方法要求传入的是每条事件轨迹上所有事件的实际x坐标集合，而非0/1二值矩阵。直接传入原始DataFrame时，01矩阵不符合输入结构要求，会被内部解析为无维度的空数组，触发“axis -1 is out of bounds for array of dimension 0”的报错；即使转成numpy数组消掉报错，程序也会把数值本身（0或1）当成事件的位置坐标，完全忽略“行索引对应x位置、值为1标记事件”的规则，最终输出图面严重错位。
• 参数配置不符合需求：
  • 原代码设置lineoffsets=1，会把所有事件线强制堆叠在同一个y值位置，没有为每一列分配独立的纵向偏移，自然无法实现每列对应一条独立堆叠线的效果
  • 原代码设置orientation='vertical'，会将短划线设置为垂直方向（沿y轴延伸），和需要的水平短划线栅格效果方向完全相反

1. 数据格式转换：遍历DataFrame的每一列，提取该列中值为1的元素对应的行索引，组成该列对应的事件x坐标列表。比如SP3列在第3、第5行值为1，对应的事件坐标就是[3,5]
2. 纵向偏移分配：为每一列生成独立的y轴位置值，让所有事件线沿y轴依次堆叠
3. 匹配绘图参数，设置对应刻度标签完成可视化

import matplotlib.pyplot as plt
import pandas as pd

# 测试用玩具数据
df = pd.DataFrame({
    'SP1': [1.0, 0.0, 0.0, 0.0, 0.0, 0.0],
    'SP3': [0.0, 0.0, 0.0, 1.0, 0.0, 1.0],
    'SP5': [0.0, 0.0, 0.0, 0.0, 1.0, 0.0],
    'SP7': [0.0, 1.0, 0.0, 0.0, 0.0, 0.0],
    'SP9': [0.0, 0.0, 0.0, 0.0, 1.0, 1.0],
    'SP11': [0.0, 0.0, 0.0, 0.0, 0.0, 1.0]
})

# 转换为eventplot要求的输入格式：每列对应一条线，存储该线上所有事件的x坐标
event_x_list = [df.index[col == 1.0].tolist() for col in df.T.values]
# 为每列分配独立的y轴偏移，实现纵向堆叠
y_positions = list(range(df.shape[1]))

plt.eventplot(
    event_x_list,
    colors='black',
    lineoffsets=y_positions,
    linelengths=0.8,
    orientation='horizontal'
)

# 匹配y轴标签为原列名，方便一一对应
plt.yticks(ticks=y_positions, labels=df.columns)
plt.xlabel('DataFrame行索引')
plt.show()

内容的提问来源于stack exchange，提问作者cjstevens