空白列表到熱圖

Question

我需要從從 HTML 工具中提取的一些數據中制作一個熱圖樣式 plot （數據是均勻分離的，所以我認為繪圖方法並不重要）。 不幸的是，該工具以 a.xyz 文件的形式將所有數據點作為 3 行空格分隔列表提供。 例如...

-1 -1 .2
-1  0 .5
-1  1 .9
-1  2 .7
0  -1  0
0   0 .8
0   1 .1
0   2 .9
1  -1 .5
1   0 .4
1   1 .6
1   2 .7

我懷疑最好的方法是將文件放入 dataframe，進行爭論，然后使用sns.heatmap()或某種方法進行繪圖。 為此，我目前首先將文件重新保存到.csv，然后使用pandas.read_csv()導入（我覺得必須有更 Python 的方式來完成這些步驟）。 我已經使用 pandas.unique()來獲取列和索引值。 現在，我不知道如何正確構建和填充新的 dataframe。

到目前為止的解決方案：

import pandas as pd
import seaborn as sns

header = ['x', 'y', 'value']
df_demo = pd.read_csv(r"C:\MyLocation\\"
                      +'democsv'+'.csv', header = None, delim_whitespace=True)
df_demo.columns = header

df_pivot = df_demo.pivot(index='y',columns='x', values='value')

sns.heatmap(df_pivot)

Output

Output 來自上述代碼

我願意以其他方式解決這個問題，並提前感謝您的幫助。

Answer 1

使用pivot ：

df = pd.read_csv('file.csv', names=header, delim_whitespace=True)

df_pivot = df.pivot(index='x',columns='y', values='value')


import seaborn as sns
sns.heatmap(df_pivot)

Output：