代码之家 › 专栏 › 技术社区 › NicoH

每个元素沿一个轴的numpy数组位置

numpy arrays python

NicoH · 技术社区 · 3 年前

比如说,我有一个numpy 2D阵列

a = np.arange(1,7).reshape(2,3)
array([[1, 2, 3],
       [4, 5, 6]])

我想要一个数组,每个元素在一个轴上的位置如下:

array([[0, 1, 2],
       [0, 1, 2]])

我需要这个来证明我的论点 matplotlib.pyplot 是的 bar3d . 有没有简单的方法可以做到这一点?

0 回复 | 直到 3 年前

Dani Mesejo 3 年前

import numpy as np

a = np.arange(1,7).reshape(2,3)

n_rows, n_cols = a.shape
res = np.tile(np.arange(n_cols), (n_rows, 1))
print(res)

输出

[[0 1 2]
 [0 1 2]]

推荐文章

July · 如何定义数字间隔,然后四舍五入

1 年前

Community wiki · 对象名称前的单下划线和双下划线的含义是什么?

1 年前

Brian Johnson · 为什么在Python中列出字典列表会引发TypeError?[已关闭]

1 年前

user026 · 如何根据特定窗口的平均值(行数)创建新列?

1 年前

Ashok Shrestha · 需要追踪特定的颜色线并获取坐标

1 年前

Nicote Ool · 在FastApi和Vue3中获得422

1 年前

NeoExceptCodeBad · 如果我有很多垂直线,我如何找到它们的边缘?

1 年前

Abdulaziz · 如何对集合内的列表进行排序[重复]

1 年前

user2743931 · 带有src目录的Python setup.py

1 年前

asmgx · 为什么合并数据帧不能按照python中的预期方式工作

1 年前