首页
学习
活动
专区
圈层
工具
发布
首页
学习
活动
专区
圈层
工具
MCP广场
社区首页 >问答首页 >如何绘制轴不是相同形状的三维数据?

如何绘制轴不是相同形状的三维数据?
EN

Stack Overflow用户
提问于 2022-01-12 08:53:13
回答 1查看 111关注 0票数 0

我有这样的数据:

代码语言:javascript
运行
复制
[
    [1, 2, 3, 4, 5],  #x
    [6, 7, 8, 9, 0],  #y
    [[1, 2, 3], [0, 2, 3], [1, 7, 3], [1, 2, 9], [1, 1, 3]]  #z is 3 values for each one value in x and y
]

如何在matplotlib中绘制/可视化这样的数据?

我尝试了以下几点,但都没有成功:

代码语言:javascript
运行
复制
ax.scatter3D(data[0], data[1], data[2], c=data[2], cmap='Greens');
# and
ax.plot3D(data[0], data[1], data[2], 'gray')

它给我带来了错误:

代码语言:javascript
运行
复制
# first line ==> ValueError: shape mismatch: objects cannot be broadcast to a single shape
# second line ==> ValueError: input operand has more dimensions than allowed by the axis remapping
EN

Stack Overflow用户

回答已采纳

发布于 2022-01-12 09:26:23

您需要重新调整数据以展开xy以匹配z

使用纯python:

代码语言:javascript
运行
复制
from itertools import product
a,b,c = zip(*(e for x,y,z in zip(*data) for e in product([x],[y],z)))

import matplotlib.pyplot as plt
fig = plt.figure()
ax = fig.add_subplot(projection='3d')

ax.scatter3D(a, b, c, c=c, cmap='Greens');

使用numpy:

代码语言:javascript
运行
复制
import numpy as np
x, y = np.repeat(np.array(data[:2]), 3, axis=1)
z = data[2]

import matplotlib.pyplot as plt
fig = plt.figure()
ax = fig.add_subplot(projection='3d')
ax.scatter3D(x, y, z, c=z, cmap='Greens')

产出:

票数 1
EN
查看全部 1 条回答
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/70678651

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档