3

Matplotlibを使用して曲線のファミリーをグラフ化しようとしています。を使用してデータを直接グラフ化し、を使用しscatter()てフィットライン（scipyからの最小二乗）をプロットしていplot()ます。事前にいくつのデータセットがあるのか​​、限界などはわかりません。

1つのデータセットのすべてが一致するように、これらの線と点の色を循環できるようにする必要があります。プロットは、内部のデフォルトを使用して色を回転させ、スキャッターはすべて1つの色として出力されます。データセットは互いに接近する可能性があるため、どのポイントがどのフィットラインに近いかが明確であると仮定して、十分ではなく、手動で色を作成する曲線の数がわからないためです。選択はスケーラブルではありません。

さらに、これらは曲線のファミリーであるため（トランジスタプロットを考えてください）、曲線に関連するラベルを表示できる必要があります。私がやりたいのは、フィットライン自体に情報を書き込むことです。

誰かがこれらのいずれかへの良い方法を知っていますか？

4

2 に答える 2

13

これはあなたのすべての質問に答えようとします。以下のコードは、最大7色を循環させます。さらに必要な場合は、別の回答に示されているように、より洗練されたジェネレーターを作成する必要があります。

import numpy as np
from matplotlib import pyplot as plt

def get_color():
    for item in ['r', 'g', 'b', 'c', 'm', 'y', 'k']:
        yield item

x = 0.3 * np.array(range(40))

color = get_color()

for group in range(5):
    # generates a collection of points
    y = np.exp2(x + 0.5 * group)
    # fit to a polynomial
    z = np.polyfit(x, y, 6)
    p = np.poly1d(z)

    acolor = next(color)
    
    plt.scatter(x, y, color=acolor, marker='o')
    plt.plot(x, p(x), acolor + '-', label=str(group))

plt.legend()
plt.xlim((0, 15))
plt.show()

ここに画像の説明を入力してください

上記のコードのジェネレーターは、この例では少しやり過ぎですが、より複雑な計算の構造を提供します。数色だけが必要な場合は、単純なイテレータを使用できます

>>> color = iter(list_of_colors)
>>> acolor = next(color)

無限に循環する必要がある場合は、次を使用できますitertools.cycle。

>>> from itertools import cycle
>>> color = cycle(['r', 'g', 'b', 'c', 'm', 'y', 'k'])
>>> next(color)
'r'
>>>

編集： n個の異なる色を取得するためのいくつかのオプションがあります。前に示したように、他の回答で示された方法を使用してジェネレーターを使用できます。たとえばget_color、別のジェネレータに置き換えます。

import colorsys
import numpy as np
from matplotlib import pyplot as plt

def get_color(color):
    for hue in range(color):
        hue = 1. * hue / color
        col = [int(x) for x in colorsys.hsv_to_rgb(hue, 1.0, 230)]
        yield "#{0:02x}{1:02x}{2:02x}".format(*col)
    
x = 0.3 * np.array(range(40))

color = get_color(15)

for group in range(15):
    # generates a collection of points
    y = np.exp2(x + 0.5 * group)
    # fit to a polynomial
    z = np.polyfit(x, y, 6)
    p = np.poly1d(z)

    acolor = next(color)

    plt.scatter(x, y, color=acolor, marker='o')
    plt.plot(x, p(x), color=acolor, linestyle='dashed', label=str(group))

plt.legend()
plt.xlim((0, 15))
plt.show()

あなたは15の異なる色を手に入れます。

ここに画像の説明を入力してください

ただし、同様の色は連続していて、良好な解像度/コントラストが得られません。次のコマンドで色相値をスキップすることで、コントラストを上げることができます。

for hue in range(0, color*3, 3):

多くの線を描くときのもう1つの問題は、凡例です...

于 2012-04-21T06:53:49.263 に答える
2

すべてを手動で定義せずに、任意の数の行をサポートしながら、複数の行に同じ色を付けたいという同様のケースがあります。これは私が色を生成するために思いついた関数です：

import colorsys

def get_colors(i, total):
    hue = i*(1.0/total)
    dark = [int(x) for x in colorsys.hsv_to_rgb(hue, 1.0, 100)]
    light = [int(x) for x in colorsys.hsv_to_rgb(hue, 1.0, 230)]
    return "#{0:02x}{1:02x}{2:02x}".format(*dark), "#{0:02x}{1:02x}{2:02x}".format(*light)

ご覧のとおりtotal、暗いバージョンと明るいバージョンで最大距離の色を生成します。

于 2012-04-20T22:27:37.417 に答える