다양한 색상으로 선 그리기

matplotlib에서는 여러 가지 접근 방식을 통해 고유한 색상 세그먼트로 선을 그릴 수 있습니다. 선택은 플롯할 선분의 수에 따라 다릅니다.

작은 수의 선분

궤적을 그릴 때와 같이 몇 개의 선분만 필요한 경우 다음을 고려하십시오.

import numpy as np
import matplotlib.pyplot as plt

# Generate random data
xy = (np.random.random((10, 2)) - 0.5).cumsum(axis=0)

fig, ax = plt.subplots()

# Plot each line segment with a unique color
for start, stop in zip(xy[:-1], xy[1:]):
    x, y = zip(start, stop)
    ax.plot(x, y, color=plt.cm.gist_ncar(np.random.random()))

plt.show()

많은 수의 선 세그먼트

많은 수의 선 세그먼트를 처리할 때 더 효율적인 방법은 LineCollection을 활용하는 것입니다.

import numpy as np
import matplotlib.pyplot as plt
from matplotlib.collections import LineCollection

# Generate random data
xy = (np.random.random((1000, 2)) - 0.5).cumsum(axis=0)

# Reshape data for compatibility with LineCollection
xy = xy.reshape(-1, 1, 2)
segments = np.hstack([xy[:-1], xy[1:]])

fig, ax = plt.subplots()

# Create a LineCollection with randomly assigned colors
coll = LineCollection(segments, cmap=plt.cm.gist_ncar)
coll.set_array(np.random.random(xy.shape[0]))

# Add the LineCollection to the plot
ax.add_collection(coll)
ax.autoscale_view()

plt.show()

두 방법 모두 Matplotlib 문서를 참조하여 선택한 컬러맵을 변경할 수 있습니다.