首页
学习
活动
专区
圈层
工具
发布
首页
学习
活动
专区
圈层
工具
MCP广场
社区首页 >问答首页 >一维元组到二维numpy数组

一维元组到二维numpy数组
EN

Stack Overflow用户
提问于 2019-07-12 02:08:17
回答 2查看 465关注 0票数 3

将元组转换为具有以下条件的Numpy矩阵:

  • 数组的形状应该是len(tuple) x len(tuple) (方阵)。
  • 数组中由(index of the element in the tuple, the value of the element in the tuple)指定的位置的元素应该是一个。

例如,我有一个随机元组,如下所示:

代码语言:javascript
运行
复制
# index means row ,value means col
(2,0,1)

我使用两个循环将这个元组转换为Numpy数组:

代码语言:javascript
运行
复制
def get_np_represent(result):
    two_D = []
    for row in range(len(result)):
        one_D = []
        for col in range(len(result)):
            if result[row] == col:
                one_D.append(1)
            else:
                one_D.append(0)
        two_D.append(one_D)
    return np.array(two_D)

产出:

代码语言:javascript
运行
复制
array([[0, 0, 1],
       [1, 0, 0],
       [0, 1, 0]])

但是我有一千万这样的元组,有更快的方法吗?

EN

Stack Overflow用户

回答已采纳

发布于 2019-07-12 02:23:47

像这样吗?操纵矩阵的速度比循环快得多。

代码语言:javascript
运行
复制
import numpy as np

t = (2, 0, 1)
x = np.zeros([len(t),len(t)])

for i,v in enumerate(t):
    x[i, v] = 1

print(x)

产出:

代码语言:javascript
运行
复制
[[0. 0. 1.]
 [1. 0. 0.]
 [0. 1. 0.]]
票数 3
EN
查看全部 2 条回答
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/56999369

复制
相关文章

相似问题

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