다양한 색상으로 선 그리기

문제 설명

latt와 lont라는 두 개의 목록이 주어졌을 때 목표는 각 항목이 있는 단일 선을 그리는 것입니다. 10개의 연속된 포인트 세그먼트는 다른 색상으로 표시됩니다.

해결책

라인 세그먼트 수 제한

라인 세그먼트 수가 적은 경우 , 예를 들어 10 이하인 경우 간단한 접근 방식은 루프를 사용하여 각 세그먼트를 고유한 색상으로 플롯하는 것입니다.

import numpy as np
import matplotlib.pyplot as plt

# Generate random colors
def uniqueish_color():
    return plt.cm.gist_ncar(np.random.random())

# Plot the line segments
xy = (np.random.random((10, 2)) - 0.5).cumsum(axis=0)
fig, ax = plt.subplots()
for start, stop in zip(xy[:-1], xy[1:]):
    x, y = zip(start, stop)
    ax.plot(x, y, color=uniqueish_color())
plt.show()

많은 수의 선 세그먼트

많은 수의 선 세그먼트의 경우 루프를 사용하면 속도가 느려질 수 있습니다. 대신 LineCollection 객체를 생성하세요.

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

# Generate the line segments
xy = (np.random.random((1000, 2)) - 0.5).cumsum(axis=0)
xy = xy.reshape(-1, 1, 2)
segments = np.hstack([xy[:-1], xy[1:]])

# Create a LineCollection object
fig, ax = plt.subplots()
coll = LineCollection(segments, cmap=plt.cm.gist_ncar)

# Set the color array
coll.set_array(np.random.random(xy.shape[0]))

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

# Display the plot
plt.show()

색상 선택

두 접근 방식 모두 "gist_ncar" 색상맵을 사용하여 고유한 색상을 생성합니다. 다른 컬러맵 옵션에 대해서는 이 페이지를 참조하십시오: http://matplotlib.org/examples/color/colormaps_reference.html