使用其他数组初始化数组

时间:2019-03-11 13:54:05

标签: python

我有一个代码如下:-

.....
path = 'path_to_csv_file';
file=open(path, "r")
reader = csv.reader(file)
y=np.empty((7000,1))
j=0
for line in reader:
    y[j]=line[0]
    j+=1

....

targets=np.zeros([7000,1,10])

现在,在第一个目标数组中,我希望第y[0]个索引存储1(y[0]存储来自0-9的整数)。为此,我写道:-

targets[0,0,y[0]]=1

但是我得到一个错误:-

IndexError: arrays used as indices must be of integer (or boolean) type

当我打印y[0]时,我得到:-

[6.]

作为输出。我认为这不是整数,所以这可能是我的错误的根源,但我不知道如何解决。任何帮助将不胜感激。谢谢!

1 个答案:

答案 0 :(得分:1)

您尝试过使用dtype=int吗?

y=np.empty((7000,1), dtype=int)
...
targets=np.zeros(([7000,1,10]), dtype=int)

您可以查看有关numpty.emptynumpty.zeros用法的文档的更多信息