将x值添加到仅包含y值的numpy数组

时间:2019-03-22 14:37:08

标签: python numpy

我有一个仅在一列中包含y值的数据集。我想插入一列x值,该值必须为0、10、20、30 ...,只要是数组。

我该怎么办?

2 个答案:

答案 0 :(得分:0)

这应该做到:

myY = [1, 4, 3, 6, 8, 4]
myX = [i * 10 for i,y in enumerate(myY)]

print(myX)  # -> [0, 10, 20, 30, 40, 50]

修改

如果您正在使用Numpy,则可以执行以下操作:

myY = np.array([1, 4, 3, 6, 8, 4])
myData = np.vstack((myY, [i * 10 for i,y in enumerate(myY)]))

何时

答案 1 :(得分:0)

假设您的y就像

y:

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

你可以做

x = np.arrange(0, y.shape[0]*10, 10)

跟着

x = x.reshape(-1,1)

x:

array([[ 0],
       [10],
       [20],
       [30],
       [40]])

现在使用

堆叠它们
np.hstack((x,y))

结果:

array([[ 0,  0],
       [10,  1],
       [20,  2],
       [30,  3],
       [40,  4]])