多个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()

湾使用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()

Answer 2

如果您将fig, axes = plt.subplots(2, 2, sharex=True, sharey=True)更改为fig, ax = plt.subplots()。然后制作一个您想要使用的颜色列表，并使用color=colors[i]：

将其传递给您的绘图

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"""

colors = ['red','green','blue','black']

fig, ax = plt.subplots()
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=ax, color=colors[i])

这导致以下图表：