使用pyplot创建多行图表

Question

将“轮次”作为X轴......实现这一目标的最佳方式是什么......我正在使用read_csv从CSV文件中读取数据，并使用pyplots获取结果......有人能帮忙吗

我尝试使用pyplots来绘制图形，但结果有问题......我期望为每个客户得到10条线图，每条线图在特定轮次下的值对应于'labels'列中的值。
例如，在第0轮，客户0的标签是0......
X轴应该是轮次，Y轴应该是'labels'列中所有唯一值的集合。

Tagir Valeev · Answer

尝试以下脚本，它可能会工作，或者至少能提供一个基本的思路 🙂

同时，也可以参考这个问题：
在matplotlib中绘制多条折线图

import matplotlib.pyplot as plt

# 定义所有客户的标签数据
all_labels = [
    [0, -1, -1, 1, -1, 0, 1, -1, 1,  0],
    [0, 1, 0, -1, -1, 1, 1, -1, 0,  1],
    [1, -1, -1, 1, -1, 0, -1, -1, 1,  1],
    [-1, 0, 1, 1, -1, -1, 1, 1, -1,  0],
    [1, 1, 0, -1, -1, 0, 1, -1, 1,  -1]
]

# 将所有标签数据按客户端重新组织
clients2labels = [[round[client_no] for round in all_labels] for client_no in range(len(all_labels[0]))]

# X轴为轮次
x = [i for i in range(len(all_labels))]

# 遍历每个客户端的标签并绘制折线图
for labels in clients2labels:
    plt.plot(x, labels)

# 设置X轴和Y轴的标签以及图表标题
plt.xlabel("轮次")
plt.ylabel("标签")
plt.title('各客户端情况')
plt.show()

assylias · Answer

您是否正在使用 pandas 中的 read_csv 函数来读取数据？如果是的话，您可以直接利用 matplotlib.pyplot 从 pandas 绘制图表：

import pandas as pd
import matplotlib.pyplot as plt

# 使用 pandas 读取 CSV 文件
df = pd.read_csv('你的CSV文件路径.csv')

# 如果你想在同一个图表上展示所有数据：
df.plot()

# 然而通常我们对图表有更多的控制需求
# 创建一个 ax 对象以实现更多自定义
# 下面是如果每条线都单独显示在一个子图上的情况
fig, ax = plt.subplots(nrows=1, ncols=1)  # 这里假定你想要一个一行一列的布局

# 遍历所有列进行绘图
columns = df.columns
for col in columns:
    df[col].plot(ax=ax)

# 设置坐标轴标签
ax.set_xlabel("X轴")
ax.set_ylabel("Y轴")

如果您想在同一画布上放置多个子图，改动也很小，但您可能会发现这个页面有帮助：matplotlib 子图官方文档

对于 pandas 的绘图选项，这个页面或许也能提供有用信息：DataFrame.plot() 官方档