使用 CSV 的 Python 散点图答案

【问题标题】：Python Scatter Plot using CSV使用 CSV 的 Python 散点图
【发布时间】：2020-10-21 03:40:48
【问题描述】：

使用 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()

【问题讨论】：

标签： python scatter-plot

【解决方案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();

输出图：

【讨论】：