提升数据可视化效果 Matplotlib图表颜色设置技巧详解 从基础配色到高级应用全方位掌握图表美化方法让你的数据展示更加专业直观

威震华夏关云长

引言

Matplotlib是Python中最流行的数据可视化库之一，而颜色是数据可视化中至关重要的元素。恰当的颜色设置不仅能使图表更加美观，还能提高数据的可读性和传达效果。本文将全面介绍Matplotlib中的颜色设置技巧，从基础到高级，帮助读者掌握图表美化的方法，使数据展示更加专业直观。

基础颜色设置

Matplotlib提供了多种方式来指定颜色，最基础的方法包括使用颜色名称、十六进制值、RGB元组等。

颜色名称

Matplotlib支持多种基本颜色名称，如’red’、’green’、’blue’等。这些名称直观易用，适合快速设置颜色。

2. import matplotlib.pyplot as plt
3. import numpy as np
5. # 创建简单数据
6. x = np.linspace(0, 10, 100)
7. y = np.sin(x)
9. # 使用颜色名称设置线条颜色
10. plt.figure(figsize=(10, 6))
11. plt.plot(x, y, color='red', linewidth=2, label='红色线条')
12. plt.plot(x, np.cos(x), color='blue', linewidth=2, label='蓝色线条')
13. plt.title('使用颜色名称的示例')
14. plt.xlabel('X轴')
15. plt.ylabel('Y轴')
16. plt.legend()
17. plt.grid(True)
18. plt.show()

复制代码

十六进制颜色码

十六进制颜色码提供了更精确的颜色控制，格式为’#RRGGBB’，其中RR、GG、BB分别是红色、绿色和蓝色的十六进制值。

2. import matplotlib.pyplot as plt
3. import numpy as np
5. # 创建简单数据
6. x = np.linspace(0, 10, 100)
7. y = np.sin(x)
9. # 使用十六进制颜色码设置线条颜色
10. plt.figure(figsize=(10, 6))
11. plt.plot(x, y, color='#FF5733', linewidth=2, label='橙红色线条')
12. plt.plot(x, np.cos(x), color='#33FF57', linewidth=2, label='绿色线条')
13. plt.title('使用十六进制颜色码的示例')
14. plt.xlabel('X轴')
15. plt.ylabel('Y轴')
16. plt.legend()
17. plt.grid(True)
18. plt.show()

复制代码

RGB/RGBA元组

RGB/RGBA元组提供了更灵活的颜色控制方式，RGB值范围在0到1之间，A表示透明度。

2. import matplotlib.pyplot as plt
3. import numpy as np
5. # 创建简单数据
6. x = np.linspace(0, 10, 100)
7. y = np.sin(x)
9. # 使用RGB/RGBA元组设置线条颜色
10. plt.figure(figsize=(10, 6))
11. plt.plot(x, y, color=(1, 0, 0), linewidth=2, label='红色线条')  # RGB
12. plt.plot(x, np.cos(x), color=(0, 0, 1, 0.5), linewidth=2, label='半透明蓝色线条')  # RGBA
13. plt.title('使用RGB/RGBA元组的示例')
14. plt.xlabel('X轴')
15. plt.ylabel('Y轴')
16. plt.legend()
17. plt.grid(True)
18. plt.show()

复制代码

HTML颜色名称

除了基本颜色名称，Matplotlib还支持HTML颜色名称，如’cornflowerblue’、’tomato’等。

2. import matplotlib.pyplot as plt
3. import numpy as np
5. # 创建简单数据
6. x = np.linspace(0, 10, 100)
7. y = np.sin(x)
9. # 使用HTML颜色名称设置线条颜色
10. plt.figure(figsize=(10, 6))
11. plt.plot(x, y, color='cornflowerblue', linewidth=2, label='矢车菊蓝')
12. plt.plot(x, np.cos(x), color='tomato', linewidth=2, label='番茄红')
13. plt.title('使用HTML颜色名称的示例')
14. plt.xlabel('X轴')
15. plt.ylabel('Y轴')
16. plt.legend()
17. plt.grid(True)
18. plt.show()

复制代码

颜色循环和调色板

在绘制多条线或多个数据系列时，Matplotlib会自动使用颜色循环。了解和自定义颜色循环对于创建协调一致的图表至关重要。

默认颜色循环

Matplotlib有一个默认的颜色循环，它会为每个新的数据系列自动分配不同的颜色。

2. import matplotlib.pyplot as plt
3. import numpy as np
5. # 创建简单数据
6. x = np.linspace(0, 10, 100)
8. # 使用默认颜色循环绘制多条线
9. plt.figure(figsize=(10, 6))
10. for i in range(1, 6):
11.     plt.plot(x, np.sin(x) + i, label=f'线条 {i}')
13. plt.title('默认颜色循环示例')
14. plt.xlabel('X轴')
15. plt.ylabel('Y轴')
16. plt.legend()
17. plt.grid(True)
18. plt.show()

复制代码

获取和修改默认颜色循环

我们可以获取当前的默认颜色循环，并根据自己的需求进行修改。

2. import matplotlib.pyplot as plt
3. import numpy as np
5. # 获取默认颜色循环
6. default_colors = plt.rcParams['axes.prop_cycle'].by_key()['color']
7. print("默认颜色循环:", default_colors)
9. # 修改默认颜色循环
10. plt.rcParams['axes.prop_cycle'] = plt.cycler(color=['#1f77b4', '#ff7f0e', '#2ca02c', '#d62728', '#9467bd'])
12. # 创建简单数据
13. x = np.linspace(0, 10, 100)
15. # 使用修改后的颜色循环绘制多条线
16. plt.figure(figsize=(10, 6))
17. for i in range(1, 6):
18.     plt.plot(x, np.sin(x) + i, label=f'线条 {i}')
20. plt.title('修改后的颜色循环示例')
21. plt.xlabel('X轴')
22. plt.ylabel('Y轴')
23. plt.legend()
24. plt.grid(True)
25. plt.show()

复制代码

使用自定义调色板

除了修改默认颜色循环，我们还可以创建自定义调色板，在特定图表中使用。

2. import matplotlib.pyplot as plt
3. import numpy as np
4. from matplotlib.colors import ListedColormap
6. # 创建自定义调色板
7. custom_palette = ['#FF5733', '#33FF57', '#3357FF', '#F3FF33', '#FF33F3']
9. # 创建简单数据
10. x = np.linspace(0, 10, 100)
12. # 使用自定义调色板绘制多条线
13. plt.figure(figsize=(10, 6))
14. for i, color in enumerate(custom_palette):
15.     plt.plot(x, np.sin(x) + i, color=color, label=f'线条 {i+1}')
17. plt.title('自定义调色板示例')
18. plt.xlabel('X轴')
19. plt.ylabel('Y轴')
20. plt.legend()
21. plt.grid(True)
22. plt.show()

复制代码

使用seaborn调色板

Seaborn库提供了许多美观的调色板，可以与Matplotlib结合使用。

2. import matplotlib.pyplot as plt
3. import numpy as np
4. import seaborn as sns
6. # 设置seaborn调色板
7. sns.set_palette("husl", 8)
9. # 创建简单数据
10. x = np.linspace(0, 10, 100)
12. # 使用seaborn调色板绘制多条线
13. plt.figure(figsize=(10, 6))
14. for i in range(1, 9):
15.     plt.plot(x, np.sin(x) + i, label=f'线条 {i}')
17. plt.title('使用Seaborn调色板示例')
18. plt.xlabel('X轴')
19. plt.ylabel('Y轴')
20. plt.legend()
21. plt.grid(True)
22. plt.show()

复制代码

分类数据的颜色处理

当处理分类数据时，为每个类别分配独特且易于区分的颜色非常重要。Matplotlib提供了多种方法来处理分类数据的颜色。

为分类数据设置颜色

在绘制柱状图、饼图等图表时，我们可以为每个类别设置不同的颜色。

2. import matplotlib.pyplot as plt
3. import numpy as np
5. # 创建分类数据
6. categories = ['类别A', '类别B', '类别C', '类别D', '类别E']
7. values = [23, 45, 56, 78, 32]
9. # 为每个类别设置颜色
10. colors = ['#FF5733', '#33FF57', '#3357FF', '#F3FF33', '#FF33F3']
12. # 绘制柱状图
13. plt.figure(figsize=(10, 6))
14. plt.bar(categories, values, color=colors)
15. plt.title('分类数据的颜色设置')
16. plt.xlabel('类别')
17. plt.ylabel('值')
18. plt.grid(True, axis='y')
19. plt.show()

复制代码

使用颜色映射处理分类数据

对于分类数据，我们可以使用离散的颜色映射（colormap）来分配颜色。

2. import matplotlib.pyplot as plt
3. import numpy as np
4. from matplotlib.cm import get_cmap
6. # 创建分类数据
7. categories = ['类别A', '类别B', '类别C', '类别D', '类别E']
8. values = [23, 45, 56, 78, 32]
10. # 获取颜色映射
11. cmap = get_cmap('tab10')
12. colors = [cmap(i) for i in np.linspace(0, 1, len(categories))]
14. # 绘制柱状图
15. plt.figure(figsize=(10, 6))
16. plt.bar(categories, values, color=colors)
17. plt.title('使用颜色映射处理分类数据')
18. plt.xlabel('类别')
19. plt.ylabel('值')
20. plt.grid(True, axis='y')
21. plt.show()

复制代码

为饼图设置颜色

饼图是另一种常见的分类数据可视化方式，颜色设置尤为重要。

2. import matplotlib.pyplot as plt
3. import numpy as np
5. # 创建分类数据
6. categories = ['类别A', '类别B', '类别C', '类别D', '类别E']
7. values = [23, 45, 56, 78, 32]
9. # 为每个类别设置颜色
10. colors = ['#FF5733', '#33FF57', '#3357FF', '#F3FF33', '#FF33F3']
12. # 绘制饼图
13. plt.figure(figsize=(10, 8))
14. plt.pie(values, labels=categories, colors=colors, autopct='%1.1f%%', startangle=90)
15. plt.title('饼图的颜色设置')
16. plt.axis('equal')  # 使饼图呈圆形
17. plt.show()

复制代码

为堆叠图设置颜色

堆叠图也是展示分类数据的常用方式，合理的颜色设置可以使数据更易于理解。

2. import matplotlib.pyplot as plt
3. import numpy as np
5. # 创建分类数据
6. categories = ['类别A', '类别B', '类别C', '类别D', '类别E']
7. years = ['2018', '2019', '2020', '2021', '2022']
8. data = np.array([
9.     [23, 45, 56, 78, 32],
10.     [34, 56, 67, 45, 23],
11.     [45, 67, 78, 56, 34],
12.     [56, 78, 89, 67, 45],
13.     [67, 89, 90, 78, 56]
14. ]).T
16. # 为每个类别设置颜色
17. colors = ['#FF5733', '#33FF57', '#3357FF', '#F3FF33', '#FF33F3']
19. # 绘制堆叠柱状图
20. plt.figure(figsize=(12, 8))
21. bottom = np.zeros(len(years))
22. for i, (category, color) in enumerate(zip(categories, colors)):
23.     plt.bar(years, data[i], bottom=bottom, color=color, label=category)
24.     bottom += data[i]
26. plt.title('堆叠柱状图的颜色设置')
27. plt.xlabel('年份')
28. plt.ylabel('值')
29. plt.legend()
30. plt.grid(True, axis='y')
31. plt.show()

复制代码

连续数据的颜色映射

对于连续数据，我们通常使用颜色映射（colormap）来表示数值的大小。Matplotlib提供了多种内置的颜色映射，也支持自定义颜色映射。

使用内置颜色映射

Matplotlib提供了多种内置的颜色映射，如’viridis’、’plasma’、’inferno’、’magma’、’cividis’等。

2. import matplotlib.pyplot as plt
3. import numpy as np
5. # 创建连续数据
6. x = np.linspace(-5, 5, 100)
7. y = np.linspace(-5, 5, 100)
8. X, Y = np.meshgrid(x, y)
9. Z = np.sin(np.sqrt(X**2 + Y**2))
11. # 使用内置颜色映射绘制等高线图
12. plt.figure(figsize=(12, 6))
14. plt.subplot(1, 2, 1)
15. contour = plt.contourf(X, Y, Z, 20, cmap='viridis')
16. plt.colorbar(contour)
17. plt.title('使用viridis颜色映射')
19. plt.subplot(1, 2, 2)
20. contour = plt.contourf(X, Y, Z, 20, cmap='plasma')
21. plt.colorbar(contour)
22. plt.title('使用plasma颜色映射')
24. plt.tight_layout()
25. plt.show()

复制代码

创建自定义颜色映射

我们可以创建自定义的颜色映射，以满足特定的可视化需求。

2. import matplotlib.pyplot as plt
3. import numpy as np
4. from matplotlib.colors import LinearSegmentedColormap
6. # 创建连续数据
7. x = np.linspace(-5, 5, 100)
8. y = np.linspace(-5, 5, 100)
9. X, Y = np.meshgrid(x, y)
10. Z = np.sin(np.sqrt(X**2 + Y**2))
12. # 创建自定义颜色映射
13. colors = [(0, 'blue'), (0.5, 'white'), (1, 'red')]
14. cmap_name = 'blue_white_red'
15. custom_cmap = LinearSegmentedColormap.from_list(cmap_name, colors)
17. # 使用自定义颜色映射绘制等高线图
18. plt.figure(figsize=(10, 8))
19. contour = plt.contourf(X, Y, Z, 20, cmap=custom_cmap)
20. plt.colorbar(contour)
21. plt.title('使用自定义颜色映射')
22. plt.show()

复制代码

使用发散颜色映射

发散颜色映射适用于有明确中心点的数据，如温度变化、偏差等。

2. import matplotlib.pyplot as plt
3. import numpy as np
5. # 创建连续数据
6. x = np.linspace(-5, 5, 100)
7. y = np.linspace(-5, 5, 100)
8. X, Y = np.meshgrid(x, y)
9. Z = X * Y  # 这个函数在正负值之间变化
11. # 使用发散颜色映射绘制等高线图
12. plt.figure(figsize=(12, 6))
14. plt.subplot(1, 2, 1)
15. contour = plt.contourf(X, Y, Z, 20, cmap='RdBu')
16. plt.colorbar(contour)
17. plt.title('使用RdBu发散颜色映射')
19. plt.subplot(1, 2, 2)
20. contour = plt.contourf(X, Y, Z, 20, cmap='seismic')
21. plt.colorbar(contour)
22. plt.title('使用seismic发散颜色映射')
24. plt.tight_layout()
25. plt.show()

复制代码

调整颜色映射的范围

我们可以调整颜色映射的范围，以突出显示数据的特定部分。

2. import matplotlib.pyplot as plt
3. import numpy as np
5. # 创建连续数据
6. x = np.linspace(-5, 5, 100)
7. y = np.linspace(-5, 5, 100)
8. X, Y = np.meshgrid(x, y)
9. Z = np.sin(np.sqrt(X**2 + Y**2))
11. # 调整颜色映射的范围
12. plt.figure(figsize=(12, 6))
14. plt.subplot(1, 2, 1)
15. contour = plt.contourf(X, Y, Z, 20, cmap='viridis')
16. plt.colorbar(contour)
17. plt.title('默认颜色映射范围')
19. plt.subplot(1, 2, 2)
20. contour = plt.contourf(X, Y, Z, 20, cmap='viridis', vmin=-0.5, vmax=0.5)
21. plt.colorbar(contour)
22. plt.title('调整后的颜色映射范围')
24. plt.tight_layout()
25. plt.show()

复制代码

高级颜色技巧

除了基本的颜色设置，Matplotlib还提供了一些高级的颜色技巧，可以进一步增强数据可视化的效果。

使用透明度

透明度（alpha）可以帮助我们处理重叠的数据，或者创建更柔和的视觉效果。

2. import matplotlib.pyplot as plt
3. import numpy as np
5. # 创建数据
6. np.random.seed(42)
7. x = np.random.normal(0, 1, 1000)
8. y = np.random.normal(0, 1, 1000)
10. # 使用透明度绘制散点图
11. plt.figure(figsize=(12, 6))
13. plt.subplot(1, 2, 1)
14. plt.scatter(x, y, alpha=0.1, color='blue')
15. plt.title('低透明度散点图')
17. plt.subplot(1, 2, 2)
18. plt.scatter(x, y, alpha=1.0, color='blue')
19. plt.title('无透明度散点图')
21. plt.tight_layout()
22. plt.show()

复制代码

创建渐变效果

渐变效果可以增强视觉吸引力，特别适用于展示变化趋势。

2. import matplotlib.pyplot as plt
3. import numpy as np
4. from matplotlib.colors import LinearSegmentedColormap
5. from matplotlib.collections import LineCollection
7. # 创建数据
8. x = np.linspace(0, 10, 100)
9. y = np.sin(x)
11. # 创建点对点线段
12. points = np.array([x, y]).T.reshape(-1, 1, 2)
13. segments = np.concatenate([points[:-1], points[1:]], axis=1)
15. # 创建颜色映射
16. cmap = plt.get_cmap('viridis')
17. norm = plt.Normalize(x.min(), x.max())
19. # 创建线段集合
20. lc = LineCollection(segments, cmap=cmap, norm=norm)
21. lc.set_array(x)
22. lc.set_linewidth(2)
24. # 绘制渐变线
25. plt.figure(figsize=(10, 6))
26. plt.gca().add_collection(lc)
27. plt.xlim(x.min(), x.max())
28. plt.ylim(y.min() - 0.1, y.max() + 0.1)
29. plt.colorbar(lc, label='X值')
30. plt.title('渐变线条效果')
31. plt.xlabel('X轴')
32. plt.ylabel('Y轴')
33. plt.grid(True)
34. plt.show()

复制代码

多色图表

在一个图表中使用多种颜色方案，可以帮助区分不同类型的数据。

2. import matplotlib.pyplot as plt
3. import numpy as np
5. # 创建数据
6. x = np.linspace(0, 10, 100)
7. y1 = np.sin(x)
8. y2 = np.cos(x)
9. y3 = np.sin(x) * np.cos(x)
11. # 创建多色图表
12. plt.figure(figsize=(12, 8))
14. # 绘制主图表
15. plt.plot(x, y1, color='#1f77b4', linewidth=2, label='sin(x)')
16. plt.plot(x, y2, color='#ff7f0e', linewidth=2, label='cos(x)')
17. plt.plot(x, y3, color='#2ca02c', linewidth=2, label='sin(x)*cos(x)')
19. # 添加填充区域
20. plt.fill_between(x, y1, alpha=0.2, color='#1f77b4')
21. plt.fill_between(x, y2, alpha=0.2, color='#ff7f0e')
23. # 添加散点标记
24. plt.scatter(x[::10], y1[::10], color='#d62728', s=50, zorder=5)
25. plt.scatter(x[::10], y2[::10], color='#9467bd', s=50, zorder=5)
27. # 添加标题和标签
28. plt.title('多色图表示例')
29. plt.xlabel('X轴')
30. plt.ylabel('Y轴')
31. plt.legend()
32. plt.grid(True)
34. plt.show()

复制代码

使用颜色强调重要数据

通过颜色对比，可以强调图表中的重要数据点。

2. import matplotlib.pyplot as plt
3. import numpy as np
5. # 创建数据
6. np.random.seed(42)
7. x = np.random.normal(0, 1, 50)
8. y = np.random.normal(0, 1, 50)
10. # 找出离中心最远的点
11. distances = np.sqrt(x**2 + y**2)
12. max_idx = np.argmax(distances)
14. # 绘制散点图，强调最远点
15. plt.figure(figsize=(10, 8))
16. plt.scatter(x, y, color='lightgray', s=100, alpha=0.7, label='普通点')
17. plt.scatter(x[max_idx], y[max_idx], color='red', s=200, edgecolor='black', linewidth=2, label='最远点')
19. # 添加连接线
20. for i in range(len(x)):
21.     if i != max_idx:
22.         plt.plot([x[max_idx], x[i]], [y[max_idx], y[i]], color='lightgray', alpha=0.3, linewidth=1)
24. # 添加标题和标签
25. plt.title('使用颜色强调重要数据')
26. plt.xlabel('X轴')
27. plt.ylabel('Y轴')
28. plt.legend()
29. plt.grid(True)
30. plt.axis('equal')
32. plt.show()

复制代码

颜色搭配原则

良好的颜色搭配不仅能使图表更加美观，还能提高数据的可读性和传达效果。以下是一些基本的颜色搭配原则。

对比原则

对比原则是指使用对比色来区分不同的数据系列，使图表更加清晰。

2. import matplotlib.pyplot as plt
3. import numpy as np
5. # 创建数据
6. x = np.linspace(0, 10, 100)
7. y1 = np.sin(x)
8. y2 = np.cos(x)
10. # 使用对比色绘制图表
11. plt.figure(figsize=(10, 6))
12. plt.plot(x, y1, color='#1f77b4', linewidth=2, label='sin(x)')
13. plt.plot(x, y2, color='#ff7f0e', linewidth=2, label='cos(x)')
15. # 添加标题和标签
16. plt.title('使用对比色区分数据系列')
17. plt.xlabel('X轴')
18. plt.ylabel('Y轴')
19. plt.legend()
20. plt.grid(True)
22. plt.show()

复制代码

和谐原则

和谐原则是指使用相近的颜色来创建和谐统一的视觉效果，适用于展示相关数据。

2. import matplotlib.pyplot as plt
3. import numpy as np
5. # 创建数据
6. x = np.linspace(0, 10, 100)
7. y1 = np.sin(x)
8. y2 = np.sin(x + np.pi/4)
9. y3 = np.sin(x + np.pi/2)
11. # 使用和谐色绘制图表
12. plt.figure(figsize=(10, 6))
13. plt.plot(x, y1, color='#1f77b4', linewidth=2, label='sin(x)')
14. plt.plot(x, y2, color='#5a9fd4', linewidth=2, label='sin(x + π/4)')
15. plt.plot(x, y3, color='#9ec7e8', linewidth=2, label='sin(x + π/2)')
17. # 添加标题和标签
18. plt.title('使用和谐色展示相关数据')
19. plt.xlabel('X轴')
20. plt.ylabel('Y轴')
21. plt.legend()
22. plt.grid(True)
24. plt.show()

复制代码

色盲友好原则

考虑到色盲用户，我们应该选择色盲友好的颜色方案，避免仅通过颜色区分数据。

2. import matplotlib.pyplot as plt
3. import numpy as np
5. # 创建数据
6. x = np.linspace(0, 10, 100)
7. y1 = np.sin(x)
8. y2 = np.cos(x)
10. # 使用色盲友好颜色绘制图表
11. plt.figure(figsize=(10, 6))
12. plt.plot(x, y1, color='#1f77b4', linewidth=2, label='sin(x)')
13. plt.plot(x, y2, color='#ff7f0e', linewidth=2, linestyle='--', label='cos(x)')
15. # 添加标题和标签
16. plt.title('色盲友好的颜色方案')
17. plt.xlabel('X轴')
18. plt.ylabel('Y轴')
19. plt.legend()
20. plt.grid(True)
22. plt.show()

复制代码

情感原则

不同的颜色会引发不同的情感反应，我们可以根据数据的性质选择相应的颜色。

2. import matplotlib.pyplot as plt
3. import numpy as np
5. # 创建数据
6. categories = ['积极', '中性', '消极']
7. values = [65, 30, 15]
9. # 根据情感选择颜色
10. colors = ['#2ca02c', '#7f7f7f', '#d62728']  # 绿色表示积极，灰色表示中性，红色表示消极
12. # 绘制柱状图
13. plt.figure(figsize=(10, 6))
14. bars = plt.bar(categories, values, color=colors)
16. # 添加数据标签
17. for bar in bars:
18.     height = bar.get_height()
19.     plt.text(bar.get_x() + bar.get_width()/2., height,
20.              f'{height}%',
21.              ha='center', va='bottom')
23. # 添加标题和标签
24. plt.title('使用颜色传达情感')
25. plt.xlabel('情感类别')
26. plt.ylabel('百分比')
27. plt.grid(True, axis='y')
29. plt.show()

复制代码

实战案例

通过一些实战案例，我们可以综合应用前面介绍的各种颜色技巧，创建专业、直观的数据可视化。

案例1：多维度数据可视化

在这个案例中，我们将创建一个多维度数据可视化，综合应用多种颜色技巧。

2. import matplotlib.pyplot as plt
3. import numpy as np
4. import pandas as pd
5. from matplotlib.colors import LinearSegmentedColormap
7. # 创建模拟数据
8. np.random.seed(42)
9. dates = pd.date_range('2020-01-01', periods=365)
10. values = np.cumsum(np.random.randn(365)) + 100
11. categories = np.random.choice(['A', 'B', 'C', 'D'], 365)
12. importance = np.random.rand(365)
14. # 创建自定义颜色映射
15. cmap = LinearSegmentedColormap.from_list('custom', ['#2ca02c', '#7f7f7f', '#d62728'])
17. # 创建图表
18. plt.figure(figsize=(15, 10))
20. # 子图1：时间序列图
21. plt.subplot(2, 2, 1)
22. plt.plot(dates, values, color='#1f77b4', linewidth=1.5)
23. plt.fill_between(dates, values, 100, where=(values > 100), color='green', alpha=0.3)
24. plt.fill_between(dates, values, 100, where=(values < 100), color='red', alpha=0.3)
25. plt.title('时间序列数据')
26. plt.xlabel('日期')
27. plt.ylabel('值')
28. plt.grid(True)
30. # 子图2：分类数据分布
31. plt.subplot(2, 2, 2)
32. category_counts = pd.Series(categories).value_counts()
33. colors = ['#1f77b4', '#ff7f0e', '#2ca02c', '#d62728']
34. plt.pie(category_counts, labels=category_counts.index, colors=colors, autopct='%1.1f%%')
35. plt.title('分类数据分布')
37. # 子图3：散点图
38. plt.subplot(2, 2, 3)
39. scatter = plt.scatter(range(len(values)), values, c=importance, cmap=cmap, alpha=0.7)
40. plt.colorbar(scatter, label='重要性')
41. plt.title('值与重要性的关系')
42. plt.xlabel('索引')
43. plt.ylabel('值')
44. plt.grid(True)
46. # 子图4：热力图
47. plt.subplot(2, 2, 4)
48. # 创建月份数据
49. monthly_data = np.zeros((12, 31))
50. for i, date in enumerate(dates):
51.     month = date.month - 1
52.     day = date.day - 1
53.     monthly_data[month, day] = values[i]
55. # 绘制热力图
56. heatmap = plt.imshow(monthly_data, cmap='viridis', aspect='auto')
57. plt.colorbar(heatmap, label='值')
58. plt.title('年度数据热力图')
59. plt.xlabel('日')
60. plt.ylabel('月')
62. plt.tight_layout()
63. plt.show()

复制代码

案例2：交互式数据探索

在这个案例中，我们将创建一个交互式数据探索界面，使用颜色来增强用户体验。

2. import matplotlib.pyplot as plt
3. import numpy as np
4. from matplotlib.widgets import Slider, Button
5. from matplotlib.colors import LinearSegmentedColormap
7. # 创建数据
8. np.random.seed(42)
9. x = np.linspace(0, 10, 100)
10. y = np.sin(x) + np.random.normal(0, 0.2, 100)
12. # 创建初始图表
13. fig, ax = plt.subplots(figsize=(10, 6))
14. plt.subplots_adjust(bottom=0.25)
16. # 初始参数
17. initial_amplitude = 1.0
18. initial_frequency = 1.0
19. initial_phase = 0.0
21. # 创建颜色映射
22. cmap = LinearSegmentedColormap.from_list('custom', ['#2ca02c', '#7f7f7f', '#d62728'])
24. # 绘制初始曲线
25. line, = ax.plot(x, y, color='#1f77b4', linewidth=2)
26. scatter = ax.scatter(x, y, c=y, cmap=cmap, alpha=0.7, s=50)
27. plt.colorbar(scatter, label='Y值')
29. # 设置图表标题和标签
30. ax.set_title('交互式数据探索')
31. ax.set_xlabel('X轴')
32. ax.set_ylabel('Y轴')
33. ax.grid(True)
35. # 创建滑块
36. ax_amplitude = plt.axes([0.25, 0.15, 0.65, 0.03])
37. ax_frequency = plt.axes([0.25, 0.1, 0.65, 0.03])
38. ax_phase = plt.axes([0.25, 0.05, 0.65, 0.03])
40. slider_amplitude = Slider(ax_amplitude, '振幅', 0.1, 2.0, valinit=initial_amplitude)
41. slider_frequency = Slider(ax_frequency, '频率', 0.1, 3.0, valinit=initial_frequency)
42. slider_phase = Slider(ax_phase, '相位', 0, 2*np.pi, valinit=initial_phase)
44. # 创建重置按钮
45. ax_reset = plt.axes([0.8, 0.01, 0.1, 0.03])
46. button_reset = Button(ax_reset, '重置')
48. # 更新函数
49. def update(val):
50.     amplitude = slider_amplitude.val
51.     frequency = slider_frequency.val
52.     phase = slider_phase.val
53.    
54.     y_new = amplitude * np.sin(frequency * x + phase) + np.random.normal(0, 0.2, 100)
55.     line.set_ydata(y_new)
56.     scatter.set_offsets(np.c_[x, y_new])
57.     scatter.set_array(y_new)
58.    
59.     fig.canvas.draw_idle()
61. # 重置函数
62. def reset(event):
63.     slider_amplitude.reset()
64.     slider_frequency.reset()
65.     slider_phase.reset()
67. # 注册更新函数
68. slider_amplitude.on_changed(update)
69. slider_frequency.on_changed(update)
70. slider_phase.on_changed(update)
71. button_reset.on_clicked(reset)
73. plt.show()

复制代码

案例3：多变量关系可视化

在这个案例中，我们将创建一个多变量关系可视化，使用颜色来表示第三个变量。

2. import matplotlib.pyplot as plt
3. import numpy as np
4. import pandas as pd
5. from matplotlib.colors import LinearSegmentedColormap
6. from mpl_toolkits.mplot3d import Axes3D
8. # 创建多变量数据
9. np.random.seed(42)
10. n = 200
11. x = np.random.normal(0, 1, n)
12. y = np.random.normal(0, 1, n)
13. z = x * y + np.random.normal(0, 0.5, n)
14. category = np.random.choice(['A', 'B', 'C'], n)
16. # 创建图表
17. fig = plt.figure(figsize=(15, 10))
19. # 子图1：散点图，颜色表示z值
20. ax1 = fig.add_subplot(2, 2, 1)
21. scatter = ax1.scatter(x, y, c=z, cmap='viridis', alpha=0.7)
22. plt.colorbar(scatter, label='Z值')
23. ax1.set_title('散点图：颜色表示Z值')
24. ax1.set_xlabel('X轴')
25. ax1.set_ylabel('Y轴')
26. ax1.grid(True)
28. # 子图2：分类散点图
29. ax2 = fig.add_subplot(2, 2, 2)
30. categories = ['A', 'B', 'C']
31. colors = ['#1f77b4', '#ff7f0e', '#2ca02c']
32. for cat, color in zip(categories, colors):
33.     mask = category == cat
34.     ax2.scatter(x[mask], y[mask], c=color, label=cat, alpha=0.7)
35. ax2.set_title('分类散点图')
36. ax2.set_xlabel('X轴')
37. ax2.set_ylabel('Y轴')
38. ax2.legend()
39. ax2.grid(True)
41. # 子图3：3D散点图
42. ax3 = fig.add_subplot(2, 2, 3, projection='3d')
43. scatter3d = ax3.scatter(x, y, z, c=z, cmap='viridis', alpha=0.7)
44. ax3.set_title('3D散点图')
45. ax3.set_xlabel('X轴')
46. ax3.set_ylabel('Y轴')
47. ax3.set_zlabel('Z轴')
49. # 子图4：气泡图
50. ax4 = fig.add_subplot(2, 2, 4)
51. # 将z值映射到气泡大小
52. sizes = 50 + 200 * (z - z.min()) / (z.max() - z.min())
53. scatter4 = ax4.scatter(x, y, s=sizes, c=z, cmap='viridis', alpha=0.7)
54. plt.colorbar(scatter4, label='Z值')
55. ax4.set_title('气泡图：大小和颜色表示Z值')
56. ax4.set_xlabel('X轴')
57. ax4.set_ylabel('Y轴')
58. ax4.grid(True)
60. plt.tight_layout()
61. plt.show()

复制代码

总结与最佳实践

通过本文的介绍，我们全面了解了Matplotlib中的颜色设置技巧，从基础到高级，从理论到实践。以下是一些总结和最佳实践建议：

颜色选择最佳实践

1. 考虑色盲用户：避免仅通过颜色区分数据，可以使用不同的线条样式、形状或标签来辅助区分。
2. 保持一致性：在同一项目或报告中，保持颜色方案的一致性，增强专业性和可读性。
3. 考虑文化差异：不同文化对颜色的理解可能不同，在国际化的项目中要注意这一点。
4. 避免过多颜色：过多的颜色会使图表显得混乱，通常建议使用不超过5-7种主要颜色。
5. 使用有意义的颜色：根据数据的性质选择颜色，如用红色表示警告，绿色表示积极等。

技术实现最佳实践

1. 使用颜色映射处理连续数据：对于连续数据，使用颜色映射比离散颜色更有效。
2. 调整透明度处理重叠数据：当数据点重叠时，适当调整透明度可以显示数据密度。
3. 结合多种视觉元素：不要仅依赖颜色，可以结合大小、形状、线条样式等元素来增强数据表达。
4. 创建自定义颜色方案：根据品牌或项目需求创建自定义颜色方案，增强识别度。
5. 测试不同显示设备：颜色在不同显示设备上可能有所不同，确保在多种设备上测试图表。

代码组织最佳实践

1. 集中管理颜色方案：将颜色方案定义为变量或常量，便于统一管理和修改。
2. 创建可重用的颜色函数：对于复杂的颜色设置，创建可重用的函数，提高代码效率。
3. 注释颜色选择理由：在代码中注释为什么选择特定的颜色，便于他人理解和维护。
4. 使用配置文件：对于大型项目，考虑使用配置文件管理颜色方案，便于灵活调整。

通过遵循这些最佳实践，我们可以创建出既美观又有效的数据可视化，使数据展示更加专业直观，更好地传达信息。

2. # 示例：颜色方案的最佳实践实现
4. import matplotlib.pyplot as plt
5. import numpy as np
6. from matplotlib.colors import LinearSegmentedColormap
8. # 1. 集中管理颜色方案
9. COLOR_SCHEME = {
10.     'primary': '#1f77b4',
11.     'secondary': '#ff7f0e',
12.     'success': '#2ca02c',
13.     'warning': '#d62728',
14.     'neutral': '#7f7f7f',
15.     'background': '#f5f5f5',
16.     'text': '#333333'
17. }
19. # 2. 创建可重用的颜色函数
20. def get_custom_colormap(colors, name='custom'):
21.     """创建自定义颜色映射"""
22.     return LinearSegmentedColormap.from_list(name, colors)
24. def set_chart_style(ax):
25.     """设置图表样式"""
26.     ax.set_facecolor(COLOR_SCHEME['background'])
27.     ax.grid(True, linestyle='--', alpha=0.7)
28.     ax.tick_params(colors=COLOR_SCHEME['text'])
29.     ax.xaxis.label.set_color(COLOR_SCHEME['text'])
30.     ax.yaxis.label.set_color(COLOR_SCHEME['text'])
31.     ax.title.set_color(COLOR_SCHEME['text'])
33. # 3. 使用颜色方案创建图表
34. def create_sample_chart():
35.     """创建示例图表"""
36.     # 创建数据
37.     x = np.linspace(0, 10, 100)
38.     y1 = np.sin(x)
39.     y2 = np.cos(x)
40.    
41.     # 创建图表
42.     fig, ax = plt.subplots(figsize=(10, 6))
43.    
44.     # 应用样式
45.     set_chart_style(ax)
46.    
47.     # 绘制数据
48.     ax.plot(x, y1, color=COLOR_SCHEME['primary'], linewidth=2, label='sin(x)')
49.     ax.plot(x, y2, color=COLOR_SCHEME['secondary'], linewidth=2, linestyle='--', label='cos(x)')
50.    
51.     # 填充区域
52.     ax.fill_between(x, y1, alpha=0.2, color=COLOR_SCHEME['primary'])
53.     ax.fill_between(x, y2, alpha=0.2, color=COLOR_SCHEME['secondary'])
54.    
55.     # 添加标题和标签
56.     ax.set_title('颜色方案最佳实践示例')
57.     ax.set_xlabel('X轴')
58.     ax.set_ylabel('Y轴')
59.     ax.legend()
60.    
61.     return fig, ax
63. # 4. 创建并显示图表
64. fig, ax = create_sample_chart()
65. plt.tight_layout()
66. plt.show()

复制代码

通过本文的学习，相信读者已经掌握了Matplotlib图表颜色设置的各种技巧，能够根据不同的数据类型和可视化需求，选择合适的颜色方案，创建出专业、直观、美观的数据可视化作品。在实际应用中，不断尝试和调整，结合具体的数据特点和展示目标，才能充分发挥颜色的作用，让数据讲述更加生动的故事。