python - 多个 csv 文件绘制在同一轴上，颜色不同

Question

考虑以下 csv 文件

import pandas as pd
from io import StringIO
from matplotlib import pylab as plt

csv1 = """x,y
0,1
1,0"""

csv2 = """x,y
0,0
1,1"""

csv3 = """x,y
.5,1
.5,0"""

csv4 = """x,y
0,.5
1,.5"""

我可以将它们全部绘制在不同的轴上

fig, axes = plt.subplots(2, 2, sharex=True, sharey=True)
for i, csv in enumerate([csv1, csv2, csv3, csv4]):
    r, c = i // 2, i % 2
    pd.read_csv(StringIO(csv)).plot.scatter('x', 'y', ax=axes[r, c])

fig.tight_layout()

但是如何在同一轴上绘制不同颜色的图形呢？

Answer 1

一个。使用 matplotlib

matplotlib 自动使用不同的颜色

fig, axes = plt.subplots()
for i, csv in enumerate([csv1, csv2, csv3, csv4]):
    df = pd.read_csv(StringIO(csv))
    axes.scatter(df.x, df.y)

fig.tight_layout()
plt.show()

b。使用 Pandas

需要定义要使用的颜色列表。

colors = ["blue", "orange", "green", "red"]
fig, axes = plt.subplots()
for i, csv in enumerate([csv1, csv2, csv3, csv4]):
    df = pd.read_csv(StringIO(csv)).plot.scatter('x', 'y', ax=axes, color=colors[i])

fig.tight_layout()
plt.show()