Matplotlib 中包含 3 个变量的散点图

Question

我有一个简单的用例：

我有 3 个变量，x、y 和 z。

我想将 z 绘制为 x 和 y 的函数，因此是 3D 绘图。

Answer 1

关于你提到的两个选项，这里的3D绘图其中一个可重用的例子z和颜色都的功能x和y：

import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D
import numpy as np

x = np.random.rand(20)
y = np.random.rand(20)
z = x*y

fig = plt.figure(figsize=(6, 6))
ax = fig.add_subplot(111, projection='3d')
ax.scatter(x, y, z,
           linewidths=1, alpha=.7,
           edgecolor='k',
           s = 200,
           c=z)
plt.show()

或者，这是一个示例，其中z是xand的函数y，但仅限于 2D。

import matplotlib.pyplot as plt
import numpy as np

x = np.random.rand(20)
y = np.random.rand(20)
z = x*y

fig = plt.figure(figsize=(6, 6))
plt.scatter(x, y,
           linewidths=1, alpha=.7,
           edgecolor='k',
           s = 200,
           c=z)
plt.show()