３変数を２Dへのプロット

Question

前提・実現したいこと

３つの変数（x,y,z）を持つデータを２次元のグラフで表そうと試みております

発生している問題・エラーメッセージ

知識が疎いせいか，グラフについて調べましたが，
2次元のグラフでは２変数，３変数のものは3Dのものしか見つけられませんでした

該当のソースコード

言語：python
使用しているもの：anaconda Jupyter Notebook

試したこと

現在３変数のデータを抽出することはできております

data = []#空のリスト作成
with open('vortex_change_program9.csv') as f:
for line in map(str.strip, f):
data.append(line.split(','))#1行ずつ追加していく

x1=data[2]
x2=x1[1:]
x3=len(x2)

y1=data[3]
y2=y1[1:]

z1=data[6]
z2=z1[1:]
z3=(z2)
print(x2)
print(y2)
print(z3)
↓出力結果
['3', '4', '5', '6', '7', '8', '9', '10', '11', '12', '13', '14']
['37', '38', '39', '40', '41', '42', '43', '44', '45', '46', '47', '48', '49']
['-0.0504979267716408', '-0.0190037358552217', '0.00128990481607616', '-0.0153453582897782', '-0.0076450458727777', '-0.0451858378946781', '-0.0266639478504658', '-0.0406735427677631', '-0.0153768770396709', '-0.00610945280641317', '-0.00500635616481304', '-0.00752125680446625']

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

データはこのようになっています
1行目は列番号を表しており，1列目は行番号を表しております．
x方向に列番号，y方向に行番号，z方向にそれぞれの値を与えたいと思っています

グラフのイメージとしては以下のようにしたいと考えております

Answer 1

とりあえず、現状のデータ形式のままではグラフを書くことができませんのでデータを

data = [[-0.0505, -0.019, 0.00129, ・・・],
        [-0.05172, -0.03713, -0.03176, ・・・],
        ・・・
        [-0.05305, -0.03148, -0.02636, ・・・, -0.01619]]

の形式に変換する必要があります。

現状のデータには不要な箇所が含まれていそうですので、不要な箇所を削除し、更にデータを文字列から**数値(float)**データに変換する必要がありそうです。

質問に記述されているコードから推測するとデータ（'vortex_change_program9.csv'）の構成は

• 3行目の２列目以降に Xのラベル値が格納されている
• 4行目の２列目以降に Yのラベル値が格納されている
• 7行目の２列目以降に 実際のデータが格納されている

と読み取れるのですが間違いないでしょうかね。

とすると、変換部のサンプルコード（実際のデータがないのであくまでも推測によるコード）は

data = []#空のリスト作成
with open('vortex_change_program9.csv') as f:
    for line in map(str.strip, f):
    data.append(line.split(','))

# ３行目Xのラベル値が格納かな？
x1=data[2]
x_label=x1[1:]

# ４行目にYのラベル値が格納かな？
y1=data[3]
y_label=y1[1:]

# データの行数をラベル値から得る
y_length = len(y_label)

# ７行目以降、上の行数分ループするし、変換したデータを d に入れる
d = []
for row in data[6:6+y_length]:
    # ２つ目からのデータを float に変換して、リスト d に入れる
    d.append(list(map(float, row[1:])))

のようになるかと思います。

あとは pcolor()やimshow()に上記で作成したデータ(d)を入れてあげると描画できるかと思います。

ax = plt.subplot()
ax.pcolor(d, cmap=plt.cm.Blues)

ax.set_xticks(np.arange(len(x_label)) + 0.5)
ax.set_xticklabels(x_label, minor=False)
ax.set_yticks(np.arange(len(y_label)) + 0.5)
ax.set_yticklabels(y_label, minor=False)

plt.show()

Comments

[youkan619]

無事に作成することが出来ました．
自分では気づけなかった点もご指摘頂き本当に助かりました．
ありがとうございます．

Answer 2

ヒートマップのことですかね。
Creating annotated heatmaps が参考になると思います。

質問にあるデータをもとにサイトのサンプルを少し変えてみました。
質問にあるデータのままだと数値ではなく文字列になってしまっているので気をつけてください。

import numpy as np
import matplotlib
import matplotlib.pyplot as plt
# sphinx_gallery_thumbnail_number = 2

# 縦軸
vegetables = ['37']

# 横軸
farmers = ['3', '4', '5', '6', '7', '8', '9', '10', '11', '12', '13', '14']

harvest = np.array([[-0.0504979267716408, -0.0190037358552217, 0.00128990481607616, -0.0153453582897782, -0.0076450458727777, -0.0451858378946781, -0.0266639478504658, -0.0406735427677631, -0.0153768770396709, -0.00610945280641317, -0.00500635616481304, -0.00752125680446625]])

fig, ax = plt.subplots()
im = ax.imshow(harvest)

# We want to show all ticks...
ax.set_xticks(np.arange(len(farmers)))
ax.set_yticks(np.arange(len(vegetables)))
# ... and label them with the respective list entries
ax.set_xticklabels(farmers)
ax.set_yticklabels(vegetables)

# Rotate the tick labels and set their alignment.
plt.setp(ax.get_xticklabels(), rotation=45, ha="right", rotation_mode="anchor")

ax.set_title("Harvest of local farmers (in tons/year)")
fig.tight_layout()
plt.show()

Comments

[youkan619]

無事にグラフを作ることが出来ました．
ありがとうございます．