用numpy为零和一创建索引

时间:2018-12-19 05:43:45

标签: python numpy

我希望创建一个numpy的形状数组,

(205,2),每个张量[1,0] x 205次应该看起来像这样。

我尝试使用np.ones([205,2])。但是,该值对于205次为[1,1],而不对于205次为[1,0]。

我是编程新手,我想在这里寻求所有大四生的帮助。我只是一个婴儿程序员。

2 个答案:

答案 0 :(得分:1)

也许有更好的方法(总有更好的方法!),但这就是我的想法:

a = np.ones((205, 2)) - np.array((0, 1))

或者:

a = np.ones((205, 2))
a[:, 1] = 0

或者:

a = np.zeros((205, 2))
a[:, 0] = 1

后两种解决方案最快。

答案 1 :(得分:0)

另一种使用np.tile()的方式:

np.tile([1,0], (205,1))

请参见here

另一种方法是使用np.broadcast_to()

np.broadcast_to([1,0], (205, 2))

请参见here

编辑:DYZ使用np.zeros()的解决方案,然后将1分配给另一部分似乎是这里最快的解决方案。