Matplotlib 颜色选择与应用：全面指南|极客教程

Matplotlib 颜色选择与应用：全面指南

Matplotlib 是 Python 中最流行的数据可视化库之一，它提供了丰富的颜色选择和应用方式，使得我们能够创建出美观、专业的图表。本文将全面介绍 Matplotlib 中的颜色选择和应用技巧，帮助你更好地掌握这一强大工具。

1. Matplotlib 中的颜色表示方法

在 Matplotlib 中，有多种方式来表示颜色。了解这些表示方法是正确使用颜色的基础。

1.1 颜色名称

Matplotlib 支持使用预定义的颜色名称来指定颜色。这些名称包括基本颜色（如 ‘red’、’blue’、’green’ 等）和更具体的颜色名称（如 ‘lightblue’、’darkgreen’ 等）。

示例代码：

import matplotlib.pyplot as plt

plt.figure(figsize=(8, 6))
plt.plot([1, 2, 3, 4], [1, 4, 2, 3], color='red', label='Red Line')
plt.plot([1, 2, 3, 4], [2, 3, 4, 1], color='blue', label='Blue Line')
plt.title('Color Names Example - how2matplotlib.com')
plt.legend()
plt.show()

Output:

Matplotlib 颜色选择与应用：全面指南

在这个示例中，我们使用 ‘red’ 和 ‘blue’ 这两个颜色名称来设置线条的颜色。这种方法简单直观，适合快速设置常见颜色。

1.2 RGB 和 RGBA 值

RGB（红绿蓝）和 RGBA（红绿蓝加透明度）是更精确的颜色表示方法。在 Matplotlib 中，可以使用 0 到 1 之间的浮点数元组来表示 RGB 或 RGBA 值。

示例代码：

import matplotlib.pyplot as plt

plt.figure(figsize=(8, 6))
plt.plot([1, 2, 3, 4], [1, 4, 2, 3], color=(0.8, 0.2, 0.1), label='RGB Line')
plt.plot([1, 2, 3, 4], [2, 3, 4, 1], color=(0.1, 0.5, 0.9, 0.5), label='RGBA Line')
plt.title('RGB and RGBA Colors - how2matplotlib.com')
plt.legend()
plt.show()

Output:

Matplotlib 颜色选择与应用：全面指南

在这个例子中，(0.8, 0.2, 0.1) 表示一种偏红色，而 (0.1, 0.5, 0.9, 0.5) 表示一种半透明的蓝色。使用 RGB 和 RGBA 值可以精确控制颜色，适合需要特定颜色的场景。

1.3 十六进制颜色代码

十六进制颜色代码是另一种常用的颜色表示方法。它以 ‘#’ 开头，后面跟着 6 个十六进制数字，分别表示红、绿、蓝的强度。

示例代码：

import matplotlib.pyplot as plt

plt.figure(figsize=(8, 6))
plt.plot([1, 2, 3, 4], [1, 4, 2, 3], color='#FF5733', label='Hex Color 1')
plt.plot([1, 2, 3, 4], [2, 3, 4, 1], color='#33FF57', label='Hex Color 2')
plt.title('Hexadecimal Colors - how2matplotlib.com')
plt.legend()
plt.show()

Output:

Matplotlib 颜色选择与应用：全面指南

这个例子中，’#FF5733′ 表示一种橙红色，而 ‘#33FF57’ 表示一种亮绿色。十六进制颜色代码广泛用于 Web 设计，因此在创建与网页风格一致的图表时特别有用。

2. Matplotlib 内置颜色循环

Matplotlib 提供了一个默认的颜色循环，当你绘制多条线或多个数据系列时，它会自动为每个系列选择不同的颜色。

示例代码：

import matplotlib.pyplot as plt

plt.figure(figsize=(10, 6))
for i in range(5):
    plt.plot([1, 2, 3, 4], [i+1, i+2, i+3, i+4], label=f'Line {i+1}')
plt.title('Default Color Cycle - how2matplotlib.com')
plt.legend()
plt.show()

Output:

Matplotlib 颜色选择与应用：全面指南

在这个例子中，我们没有明确指定颜色，Matplotlib 会自动为每条线选择不同的颜色。这个特性在绘制多个数据系列时非常有用，可以快速区分不同的数据。

3. 自定义颜色映射

颜色映射（Colormap）是一种将数值范围映射到颜色范围的方法。Matplotlib 提供了多种内置的颜色映射，也允许用户创建自定义的颜色映射。

3.1 使用内置颜色映射

Matplotlib 提供了许多预定义的颜色映射，如 ‘viridis’、’plasma’、’inferno’ 等。

示例代码：

import matplotlib.pyplot as plt
import numpy as np

x = np.linspace(0, 10, 100)
y = np.sin(x)

plt.figure(figsize=(10, 6))
plt.scatter(x, y, c=y, cmap='viridis')
plt.colorbar(label='Value')
plt.title('Viridis Colormap - how2matplotlib.com')
plt.show()

Output:

Matplotlib 颜色选择与应用：全面指南

这个例子使用 ‘viridis’ 颜色映射来可视化 sin 函数的值。颜色映射可以有效地表示数据的第三个维度（在这里是 y 值）。

3.2 创建自定义颜色映射

有时，内置的颜色映射可能不能满足特定需求。在这种情况下，我们可以创建自定义的颜色映射。

示例代码：

import matplotlib.pyplot as plt
import matplotlib.colors as mcolors
import numpy as np

# 创建自定义颜色映射
colors = ['blue', 'white', 'red']
n_bins = 100
cmap = mcolors.LinearSegmentedColormap.from_list('custom_cmap', colors, N=n_bins)

# 生成数据
x = np.linspace(-2, 2, 100)
y = np.linspace(-2, 2, 100)
X, Y = np.meshgrid(x, y)
Z = X * Y

plt.figure(figsize=(10, 8))
plt.pcolormesh(X, Y, Z, cmap=cmap)
plt.colorbar(label='Value')
plt.title('Custom Colormap - how2matplotlib.com')
plt.show()

Output:

Matplotlib 颜色选择与应用：全面指南

在这个例子中，我们创建了一个从蓝色过渡到白色再到红色的自定义颜色映射。这种自定义颜色映射可以用来强调数据中的特定范围或创建特殊的视觉效果。

4. 颜色归一化

颜色归一化是一种将数据值映射到颜色范围的技术。Matplotlib 提供了几种归一化方法，可以帮助我们更好地可视化数据。

4.1 线性归一化

线性归一化是最简单的归一化方法，它将数据线性映射到颜色范围。

示例代码：

import matplotlib.pyplot as plt
import numpy as np
from matplotlib.colors import Normalize

# 生成数据
x = np.linspace(0, 10, 100)
y = np.sin(x)

# 创建线性归一化对象
norm = Normalize(vmin=y.min(), vmax=y.max())

plt.figure(figsize=(10, 6))
plt.scatter(x, y, c=y, cmap='coolwarm', norm=norm)
plt.colorbar(label='Value')
plt.title('Linear Normalization - how2matplotlib.com')
plt.show()

Output:

Matplotlib 颜色选择与应用：全面指南

这个例子使用线性归一化来映射 sin 函数的值到 ‘coolwarm’ 颜色映射。这种方法适用于数据分布相对均匀的情况。

4.2 对数归一化

对于跨越多个数量级的数据，对数归一化通常更有效。

示例代码：

import matplotlib.pyplot as plt
import numpy as np
from matplotlib.colors import LogNorm

# 生成数据
x = np.linspace(1, 100, 100)
y = x ** 2

plt.figure(figsize=(10, 6))
plt.scatter(x, y, c=y, cmap='viridis', norm=LogNorm())
plt.colorbar(label='Value (log scale)')
plt.title('Logarithmic Normalization - how2matplotlib.com')
plt.yscale('log')
plt.show()

Output:

Matplotlib 颜色选择与应用：全面指南

在这个例子中，我们使用对数归一化来可视化二次函数。这种方法可以更好地展示跨越多个数量级的数据。

5. 颜色透明度控制

控制颜色的透明度可以帮助我们创建更复杂的可视化效果，特别是在需要叠加多层数据时。

5.1 使用 alpha 参数

大多数 Matplotlib 绘图函数都支持 alpha 参数，用于控制透明度。

示例代码：

import matplotlib.pyplot as plt
import numpy as np

x = np.linspace(0, 10, 100)
y1 = np.sin(x)
y2 = np.cos(x)

plt.figure(figsize=(10, 6))
plt.fill_between(x, y1, alpha=0.3, color='blue', label='Sin')
plt.fill_between(x, y2, alpha=0.3, color='red', label='Cos')
plt.title('Transparency with Alpha - how2matplotlib.com')
plt.legend()
plt.show()

Output:

Matplotlib 颜色选择与应用：全面指南

这个例子展示了如何使用 alpha 参数来创建半透明的填充区域。这种技术在展示重叠数据时特别有用。

5.2 颜色与透明度组合

我们还可以在颜色定义中直接包含透明度信息。

示例代码：

import matplotlib.pyplot as plt
import numpy as np

x = np.linspace(0, 10, 100)
y1 = np.sin(x)
y2 = np.cos(x)

plt.figure(figsize=(10, 6))
plt.plot(x, y1, color='blue', label='Sin')
plt.plot(x, y2, color='red', label='Cos')
plt.fill_between(x, y1, color='blue', alpha=0.2)
plt.fill_between(x, y2, color='red', alpha=0.2)
plt.title('Combined Color and Transparency - how2matplotlib.com')
plt.legend()
plt.show()

Output:

Matplotlib 颜色选择与应用：全面指南

在这个例子中，我们结合使用了实线和半透明填充区域来增强图表的可读性。

6. 颜色选择的最佳实践

选择合适的颜色对于创建有效的数据可视化至关重要。以下是一些最佳实践：

6.1 考虑色盲友好性

选择颜色时，应考虑色盲人群的需求。避免仅依赖红绿对比，而应使用蓝黄对比或明暗对比。

示例代码：

import matplotlib.pyplot as plt
import numpy as np

# 色盲友好的颜色选择
colors = ['#1f77b4', '#ff7f0e', '#2ca02c', '#d62728', '#9467bd']

x = np.linspace(0, 10, 100)
for i, color in enumerate(colors):
    plt.plot(x, np.sin(x + i), color=color, label=f'Line {i+1}')

plt.title('Color-Blind Friendly Palette - how2matplotlib.com')
plt.legend()
plt.show()

Output:

Matplotlib 颜色选择与应用：全面指南

这个例子使用了一组色盲友好的颜色，确保图表对大多数观众都清晰可辨。

6.2 使用对比度

高对比度可以提高图表的可读性。选择颜色时，应确保前景和背景之间有足够的对比。

示例代码：

import matplotlib.pyplot as plt
import numpy as np

plt.figure(figsize=(10, 6))
plt.style.use('dark_background')

x = np.linspace(0, 10, 100)
plt.plot(x, np.sin(x), color='#00ff00', linewidth=2, label='Sin')
plt.plot(x, np.cos(x), color='#ff00ff', linewidth=2, label='Cos')

plt.title('High Contrast Colors - how2matplotlib.com', color='white')
plt.legend()
plt.show()

Output:

Matplotlib 颜色选择与应用：全面指南

这个例子使用了深色背景和亮色线条，创造了高对比度的效果，使图表更易读。

6.3 使用颜色来传达信息

颜色不仅可以美化图表，还可以传达额外的信息。例如，使用颜色来表示数据的类别或强度。

示例代码：

import matplotlib.pyplot as plt
import numpy as np

categories = ['A', 'B', 'C', 'D', 'E']
values = [23, 45, 56, 78, 32]
colors = plt.cm.viridis(np.linspace(0, 1, len(categories)))

plt.figure(figsize=(10, 6))
bars = plt.bar(categories, values, color=colors)

for bar in bars:
    height = bar.get_height()
    plt.text(bar.get_x() + bar.get_width()/2., height,
             f'{height}', ha='center', va='bottom')

plt.title('Using Color to Convey Information - how2matplotlib.com')
plt.colorbar(plt.cm.ScalarMappable(cmap='cmap='viridis'), label='Value')
plt.show()

在这个例子中，我们使用颜色来表示不同类别的值的大小。这种方法可以快速传达数据的相对大小，而不仅仅依赖于柱状图的高度。

7. 高级颜色技巧

掌握了基础知识后，我们可以探索一些更高级的颜色应用技巧。

7.1 颜色循环自定义

虽然 Matplotlib 提供了默认的颜色循环，但有时我们可能需要自定义颜色循环以满足特定需求。

示例代码：

import matplotlib.pyplot as plt
import numpy as np
import cycler

custom_cycler = cycler(color=['#1f77b4', '#ff7f0e', '#2ca02c', '#d62728', '#9467bd'])

plt.figure(figsize=(10, 6))
plt.rc('axes', prop_cycle=custom_cycler)

x = np.linspace(0, 10, 100)
for i in range(5):
    plt.plot(x, np.sin(x + i), label=f'Line {i+1}')

plt.title('Custom Color Cycle - how2matplotlib.com')
plt.legend()
plt.show()

这个例子展示了如何创建和应用自定义颜色循环。这对于创建具有特定颜色主题的图表非常有用。

7.2 颜色映射的离散化

有时，我们可能需要将连续的颜色映射离散化为有限数量的颜色。

示例代码：

import matplotlib.pyplot as plt
import numpy as np
from matplotlib.colors import BoundaryNorm, ListedColormap

# 创建离散化的颜色映射
n_bins = 5
cmap = plt.get_cmap('viridis', n_bins)

# 生成数据
x = np.linspace(0, 10, 100)
y = np.sin(x)

# 创建颜色边界和规范化
bounds = np.linspace(y.min(), y.max(), n_bins + 1)
norm = BoundaryNorm(bounds, cmap.N)

plt.figure(figsize=(10, 6))
plt.scatter(x, y, c=y, cmap=cmap, norm=norm)
plt.colorbar(label='Discrete Values', ticks=bounds[:-1] + np.diff(bounds)/2)
plt.title('Discretized Colormap - how2matplotlib.com')
plt.show()

Output:

Matplotlib 颜色选择与应用：全面指南

这个例子展示了如何将连续的颜色映射离散化为固定数量的颜色。这在需要将连续数据分类或简化颜色表示时非常有用。

7.3 颜色混合

颜色混合可以创造出独特的视觉效果，特别是在处理重叠数据时。

示例代码：

import matplotlib.pyplot as plt
import numpy as np

def blend_colors(color1, color2, alpha):
    return [c1 * (1 - alpha) + c2 * alpha for c1, c2 in zip(color1, color2)]

plt.figure(figsize=(10, 6))

x = np.linspace(0, 10, 100)
y1 = np.sin(x)
y2 = np.cos(x)

color1 = [1, 0, 0]  # 红色
color2 = [0, 0, 1]  # 蓝色

for alpha in np.linspace(0, 1, 10):
    blended_color = blend_colors(color1, color2, alpha)
    plt.plot(x, y1 * (1 - alpha) + y2 * alpha, color=blended_color)

plt.title('Color Blending - how2matplotlib.com')
plt.show()

Output:

Matplotlib 颜色选择与应用：全面指南

这个例子展示了如何通过混合两种颜色来创建渐变效果。这种技术可以用来表示数据的连续变化或创建平滑的视觉过渡。

8. 颜色在不同类型图表中的应用

不同类型的图表可能需要不同的颜色应用策略。让我们看看如何在各种常见的图表类型中应用颜色。

8.1 热图（Heatmap）

热图是一种使用颜色来表示数值大小的图表类型，颜色的选择对其效果影响很大。

示例代码：

import matplotlib.pyplot as plt
import numpy as np

# 生成数据
data = np.random.rand(10, 10)

plt.figure(figsize=(10, 8))
plt.imshow(data, cmap='YlOrRd')
plt.colorbar(label='Value')
for i in range(10):
    for j in range(10):
        plt.text(j, i, f'{data[i, j]:.2f}', ha='center', va='center', color='black')
plt.title('Heatmap Example - how2matplotlib.com')
plt.show()

Output:

Matplotlib 颜色选择与应用：全面指南

这个例子使用 ‘YlOrRd’ 颜色映射创建了一个热图。热图中，颜色的变化直观地展示了数据的分布和强度。

8.2 等高线图

等高线图使用颜色来表示不同的高度或值。

示例代码：

import matplotlib.pyplot as plt
import numpy as np

x = np.linspace(-3, 3, 100)
y = np.linspace(-3, 3, 100)
X, Y = np.meshgrid(x, y)
Z = np.sin(X) * np.cos(Y)

plt.figure(figsize=(10, 8))
cs = plt.contourf(X, Y, Z, cmap='coolwarm', levels=20)
plt.colorbar(cs, label='Value')
plt.title('Contour Plot - how2matplotlib.com')
plt.show()

Output:

Matplotlib 颜色选择与应用：全面指南

这个例子使用 ‘coolwarm’ 颜色映射创建了一个等高线图。颜色的变化帮助观察者快速识别不同的值区域。

8.3 散点图

在散点图中，颜色可以用来表示额外的维度或类别。

示例代码：

import matplotlib.pyplot as plt
import numpy as np

# 生成数据
n = 1000
x = np.random.randn(n)
y = np.random.randn(n)
colors = np.random.rand(n)
sizes = 1000 * np.random.rand(n)

plt.figure(figsize=(10, 8))
plt.scatter(x, y, c=colors, s=sizes, alpha=0.5, cmap='viridis')
plt.colorbar(label='Random Value')
plt.title('Scatter Plot with Color and Size - how2matplotlib.com')
plt.show()

Output:

Matplotlib 颜色选择与应用：全面指南

这个例子创建了一个散点图，其中点的颜色和大小都用来表示额外的信息。这种方法可以在二维平面上展示四个维度的数据。

9. 颜色与数据类型的关系

选择合适的颜色方案不仅取决于美学考虑，还应考虑数据的类型和性质。

9.1 定性数据

对于定性数据（如类别），应选择明显不同的颜色。

示例代码：

import matplotlib.pyplot as plt
import numpy as np

categories = ['A', 'B', 'C', 'D', 'E']
values = [23, 45, 56, 78, 32]
colors = plt.cm.Set3(np.linspace(0, 1, len(categories)))

plt.figure(figsize=(10, 6))
plt.bar(categories, values, color=colors)
plt.title('Qualitative Data Visualization - how2matplotlib.com')
plt.show()

Output:

Matplotlib 颜色选择与应用：全面指南

这个例子使用 ‘Set3’ 颜色映射为不同类别选择了明显不同的颜色，有助于区分各个类别。

9.2 定量数据

对于定量数据，通常使用渐变色或连续的颜色映射。

示例代码：

import matplotlib.pyplot as plt
import numpy as np

x = np.linspace(0, 10, 100)
y = np.sin(x)

plt.figure(figsize=(10, 6))
plt.scatter(x, y, c=y, cmap='viridis')
plt.colorbar(label='Value')
plt.title('Quantitative Data Visualization - how2matplotlib.com')
plt.show()

Output:

Matplotlib 颜色选择与应用：全面指南

这个例子使用 ‘viridis’ 颜色映射来表示连续的定量数据，颜色的渐变反映了数值的变化。

10. 颜色与心理学

颜色不仅仅是视觉元素，它还能影响观者的情绪和理解。了解颜色心理学可以帮助我们更有效地传达信息。

10.1 情感联系

不同的颜色可以唤起不同的情感。例如，红色通常与热情或警告相关，而蓝色则给人冷静或专业的感觉。

示例代码：

import matplotlib.pyplot as plt
import numpy as np

x = np.linspace(0, 10, 100)
y1 = np.sin(x)
y2 = np.cos(x)

plt.figure(figsize=(10, 6))
plt.plot(x, y1, color='red', label='Excitement')
plt.plot(x, y2, color='blue', label='Calmness')
plt.title('Emotional Color Association - how2matplotlib.com')
plt.legend()
plt.show()

Output:

Matplotlib 颜色选择与应用：全面指南

这个例子使用红色和蓝色来分别表示兴奋和冷静的概念，展示了如何利用颜色的情感联系来增强数据的表现力。

10.2 文化差异

颜色的含义在不同文化中可能有所不同。在选择颜色时，应考虑目标受众的文化背景。

示例代码：

import matplotlib.pyplot as plt

categories = ['Western', 'Eastern']
values = [70, 30]
colors = ['red', 'green']

plt.figure(figsize=(10, 6))
plt.pie(values, labels=categories, colors=colors, autopct='%1.1f%%')
plt.title('Cultural Color Preferences - how2matplotlib.com')
plt.show()