化学のための Pythonによるデータ解析・機械学習入門で分からないことがあります。

Question

前提・実現したいこと

以下のコードでx_class = x[iris_types == sample_class]が分かりません。
xにはiris_typesがないと思うのですが。
よろしくおねがいします。

該当のソースコード

import matplotlib.pyplot as plt
import pandas as pd

variable_number_1 = 0  # 散布図における横軸の特徴量の番号 (0 から始まるため注意)
variable_number_2 = 1  # 散布図における縦軸の特徴量の番号

color_list = ['k', 'r', 'b', 'g', 'y', 'c', 'm']

dataset = pd.read_csv('iris.csv', index_col=0)
iris_types = dataset.iloc[:, 0] # あやめの種類
x = dataset.iloc[:, 1:]  # 数値の特徴量のみのデータセット

# 以下で散布図を描画します
plt.rcParams['font.size'] = 18  # 横軸や縦軸の名前の文字などのフォントのサイズ
for index, sample_class in enumerate(set(iris_types)):
    x_class = x[iris_types == sample_class]
    plt.scatter(x_class.iloc[:, variable_number_1], x_class.iloc[:, variable_number_2], c=color_list[index], label=sample_class)  # 散布図の作成

試したこと

ここに問題に対して試したことを記載してください。

補足情報（FW/ツールのバージョンなど）

ここにより詳細な情報を記載してください。

Comments

[meg_]

質問のコードを実行するとどうなりますか？

Answer 1

iris_typesはSeriesデータと思われるので、iris_types == sample_classはboolリスト(True/Falseのリスト)が返されます。
x[iris_types == sample_class]は、対応するboolリストがTrueになる値を返すことになります。

※入力データが一部でも記載されると理解しやすいので的確な回答がつきやすくなると思います。