python - matplotlib を使用して非数値を使用してプロットに色を付けるにはどうすればよいですか?

Question

私は matplotlib を使用して値をプロットしようとしていますが、プロットするポイントのクラスがいくつかあります。例

Number   class     x     y 
1        wood      0.2   9.4
3        glass     7.6   4.6

これらの異なるクラスを異なる色のプロットでエレガントに表現するにはどうすればよいですか?

Answer 1

異なるクラスのデータがある場合は、 を使用できますclasses。

import matplotlib.pyplot as plt

class DataSet(object):
    def __init__(self, name, x_data, y_data, size, color):
        self.name = name
        self.x_data = x_data
        self.y_data = y_data
        self.size = size
        self.color = color

wood = DataSet('wood', [1,2,3,4], [5,7,4,1], 300, 'red')
glass = DataSet('glass', [0,2,0,4], [2,7,4,1], 100, 'blue')
paper = DataSet('paper', [1,7,3,6], [9,9,1,9], 500, '#a5a5a5')
plastic = DataSet('plastic', [3,2,5,5], [1,5,1,8], 700, 'yellow')

categories = [wood, glass, paper, plastic]
#Here we plot all the data:
for cat in categories:
    plt.scatter(cat.x_data, cat.y_data, s=cat.size , color=cat.color, label=cat.name)

plt.legend(loc='lower right', scatterpoints=1)
plt.show()

これにより、次のことがわかります。

どちらがエレガントかもしれないし、そうでないかもしれません!