Python时间序列折线图设置不同颜色|极客笔记

Python时间序列折线图设置不同颜色

在数据可视化中，折线图是非常常见的一种图表类型，它可以展示数据随时间变化的趋势。在Python中，我们可以使用Matplotlib库来绘制折线图，并且可以通过设置不同的颜色来区分不同的数据系列。

本文将介绍如何使用Python的Matplotlib库来绘制时间序列折线图，并设置不同的颜色来区分不同的数据系列。

准备数据

首先，我们需要准备一些时间序列数据来绘制折线图。下面是一个简单的示例数据，包括3个数据系列：

import pandas as pd
import numpy as np

# 创建示例数据
dates = pd.date_range('20220101', periods=10)
data1 = np.random.randint(1, 10, 10)
data2 = np.random.randint(1, 10, 10)
data3 = np.random.randint(1, 10, 10)

df = pd.DataFrame({'date': dates, 'data1': data1, 'data2': data2, 'data3': data3})
df.set_index('date', inplace=True)

print(df)

运行以上代码，输出示例数据如下：

            data1  data2  data3
2022-01-01      3      2      3
2022-01-02      5      8      3
2022-01-03      7      6      9
2022-01-04      9      8      3
2022-01-05      4      9      1
2022-01-06      3      6      2
2022-01-07      7      4      9
2022-01-08      8      6      6
2022-01-09      6      4      4
2022-01-10      6      3      4

我们可以看到，示例数据包括3个数据系列data1、data2、data3，每个数据系列有10个数据点。

绘制折线图

接下来，我们使用Matplotlib库来绘制折线图，并设置不同的颜色来区分不同的数据系列。

import matplotlib.pyplot as plt

# 设置颜色
colors = ['red', 'green', 'blue']

# 绘制折线图
plt.figure(figsize=(10, 6))
for i, col in enumerate(df.columns):
    plt.plot(df.index, df[col], color=colors[i], label=col)

plt.title('Time Series Line Chart with Different Colors')
plt.xlabel('Date')
plt.ylabel('Value')
plt.legend()
plt.grid(True)
plt.show()

以上代码中，我们首先定义了颜色列表colors，其中包含了3种颜色。然后使用plt.plot()方法绘制折线图，通过color参数可以指定折线的颜色。最后，我们设置了图表的标题、横轴标签、纵轴标签、图例，并显示了图表。

在折线图中，每个数据系列都使用了不同的颜色，分别为红色、绿色、蓝色，便于区分不同的数据系列。