python - 如何在 Seaborn 中制作具有多个组或类别的联合图?
问题描述
我正在尝试在 Seaborn 中进行联合绘图。目标是生成所有 [x,z] 值的散点图,并通过 [cat] 对这些值进行颜色编码,并获得这两个类别的分布。然后我还想要 [x,alt_Z] 的散点图和分布图,忽略 NaN 的 alt_Z 值。
使用 Python 3.7
这是一个独立的数据集和我的目标(在 Excel 中制作,因此未显示分布)。
import matplotlib.pyplot as plt
%matplotlib inline
import pandas as pd
import seaborn as sns
col1 = [1,1.5,3.1,3.4,2,-1]
col2 = [1,-3,2,8,2.5,-1.3]
col3 = [4,3,4,0.5,1,0.3]
col4 = [10,12,10,'NaN',13,'NaN']
col5 = ['A','A','A','B','A','B']
df = pd.DataFrame(list(zip(col1, col2, col3, col4, col5)),
columns =['x', 'y', 'z', 'alt_Z', 'cat'])
display(df)
下面的代码没有完成绘图并返回TypeError: The y variable is categorical, but one of ['numeric', 'datetime'] is required
。在下面的代码中,我也不知道如何按 [cat] A & B 分组,因此它显示为红色,并且只有 A 类别正在绘制。
df2 = df[['x', 'y', 'z', 'alt_Z', 'cat']]\
.melt(id_vars=['x', 'y'], value_vars=['z', 'alt_Z'])
g = sns.jointplot(data=df2, x='x', y='value', hue='variable',
palette={'z': 'black', 'alt_Z': 'red'})
解决方案
数据框的一个问题是它col4
包含整数和“NaN”。由于不存在整数的 NaN 值,pandas 将其设为对象列。将其转换为浮点数将创建一个带有NaN
数字的正确浮点列。
要创建散点图,需要两次调用sns.scatter()
:
import matplotlib.pyplot as plt
import pandas as pd
import seaborn as sns
col1 = [1, 1.5, 3.1, 3.4, 2, -1]
col2 = [1, -3, 2, 8, 2.5, -1.3]
col3 = [4, 3, 4, 0.5, 1, 0.3]
col4 = [10, 12, 10, 'NaN', 13, 'NaN']
col5 = ['A', 'A', 'A', 'B', 'A', 'B']
df = pd.DataFrame(list(zip(col1, col2, col3, col4, col5)),
columns=['x', 'y', 'z', 'alt_Z', 'cat'])
df['alt_Z'] = df['alt_Z'].astype(float)
ax = sns.scatterplot(data=df, x='x', y='alt_Z', color='black', label='alt_Z')
sns.scatterplot(data=df, x='x', y='z', hue='cat', ax=ax)
plt.show()
从这里,我们可以创建 2 个数据框:df1
包含x
和。并包含和。重命名并填充包含该字符串的列将使其类似于.z
cat
df2
x
alt_Z
alt_Z
z
cat
alt_Z
df1
然后jointplot()
可以对两个数据名的串联进行操作:
df1 = df[['x', 'z', 'cat']]
df2 = df[['x', 'alt_Z']].rename(columns={'alt_Z': 'z'}).dropna()
df2['cat'] = 'alt_Z'
g = sns.jointplot(data=df1.append(df2), x='x', y='z', hue='cat', palette={'alt_Z': 'black', 'A': 'orange', 'B': 'green'})
g.ax_joint.set_xlim(-3, 6) # the default limits are too wide for these reduced test data
plt.show()
推荐阅读
- scala - 如何提取缩小类型并将其用于 Shapeless 的副产品?
- python - Numpy:ix_和ogrid有什么区别吗?
- javascript - React 无法触发 window.onmouseup
- android - Android 10 蓝牙 LE 扫描问题
- c - 例外 (9): epc1=0x4020f694 epc2=0x00000000 epc3=0x00000000 excvaddr=0x0000032d depc=0x00000000
- javascript - 在 Electron 中保存到 Windows“映射网络驱动器”的问题
- python-3.x - 反向列表不适用于我的模式?
- keras - 从 BERT 分类器中删除最后 2 层会导致“'tuple' object has no attribute 'dim'”错误。为什么?
- html - 在移动设备上查看时,CSS Grid 不能水平放置
- php - 我的 php 代码出现问题,出现错误 500