使用 CSV 的 Python 散點圖

Question

使用 Python：我正在嘗試使用 .csv 數據框制作散點圖，以可視化 FIM 錄取和出院分數。 我能夠制作散點圖，但正在努力...

制作男性散點：藍色，圓形標記（“.”），大小 100，標簽“男性（N=）”

制作女性散點：紅色，加標記（“+”），大小 100，標簽“女性（N=）”

Y = X 線：黑色虛線樣式（“--”），x 限制和 y 限制為 [0, 140] 我的圖表應該讀入我的“RIC”列，但為每個實例中的每個實例制作不同的圖表列（例如“repILE”）

它應該是這樣的：

我已經包含了到目前為止的代碼：

patient_data_df.plot.scatter(x='Admission Total FIM Score', y= 'Discharge Total FIM Score', s=50, marker= "X", color= "green")

plt.title("FIM Scatter Plot:<RIC> (N=<total>)<RIC> (N=<total>)")

#plt.show()

Answer 1

最好使用 matplotlib 而不是 pandas plot

這里示例代碼：

import pandas as pd
from matplotlib import pyplot as plt

data = pd.DataFrame([[1.1, 4.2, 'Male'], [1.2, 3.3, 'Male'], [2.3, 3.2, 'Male'], [1.4, 7.1, 'Male'], [1.3, 3.2, 'Female'], [2.3, 2.2, 'Female'], [4.2, 1.8, 'Female']], columns=['Admission Total FIM Score', 'Discharge Total FIM Score', 'Sex'])

x_name = 'Admission Total FIM Score'
y_name = 'Discharge Total FIM Score'

plt.figure(figsize=(10, 8))
plt.plot([0, 5], [0, 5], 'k--', label='No Change')
plt.plot(data[data.Sex == 'Female'][x_name], data[data.Sex == 'Female'][y_name], 'r+', label='Female (N=%i)' % len(data[data.Sex == 'Female']))
plt.plot(data[data.Sex == 'Male'][x_name], data[data.Sex == 'Male'][y_name], 'b.', label='Male (N=%i)' % len(data[data.Sex == 'Male']))
plt.title('RepILE (N=%i)' % len(data))
plt.xlabel(x_name)
plt.ylabel(y_name)
plt.legend();

輸出圖：