具有 NaN 值的气泡图

Question

我有一个这样的数据框：

x	y	height
1	1	NaN
2	5	20
3	10	100
4	7	NaN
5	6	NaN
6	12	500

我需要为 x-y 创建一个气泡图，其中“高度”作为点大小的变量（当它是 NaN 时使用其他标记）。

我正在尝试以这种方式调节高度变量，但没有成功。

fig = plt.figure()
ax = fig.add_subplot()

ax.scatter(x, y, marker='o', c='k', s=height)
ax.scatter(x[height==np.nan], y[height==np.nan], marker='^', c='r')

Answer 1

试试：

fig, ax = plt.subplots()

ax.scatter(df['x'], df['y'], marker='o', c='k', s=df['height'])

ax.scatter(df.loc[df['height'].isna(), 'x'], df.loc[df['height'].isna(), 'y'],
           marker='^', c='r')