numpy.insert()函数将数组插入错误的索引

时间:2019-02-15 19:02:51

标签: python arrays numpy multidimensional-array insert

在这里,我的代码专长于表单文本文件;并将矩阵创建为多维数组,但是问题是代码创建的数组比二维数组更多,我无法操纵,我需要二维数组,我该怎么做?

解释我的代码的算法:

代码Moto:我的代码从特定文件夹中获取值,每个文件夹包含一个用户生成的7个“ txt”文件,这样,多个文件夹将包含多个用户的多个数据。

步骤1:启动第一个for循环,并使用特定文件夹中的文件夹数量来控制它,并在变量“ path”中存储第一个文件夹的第一个路径。

第2步:使用2nd for loop打开路径并获取7个txt文件的数据。专长之后,它关闭2nd for loop并执行其余代码。

step3:将7个txt文件的数据合并到一个1d数组中。

第4步:使用获取2个文件夹的数据创建2d数组

step5(此处会出现问题):在2d数组ind inser id数组中创建一行

import numpy as np
import array as arr
import os
f_path='Result'
array_control_var=0

#for feacth directory path
for (path,dirs,file) in os.walk(f_path):
    if(path==f_path):
        continue
    f_path_1= path +'\page_1.txt'
    #Get data from page1 indivisualy beacuse there string type data exiest
    pgno_1 = np.array(np.loadtxt(f_path_1, dtype='U', delimiter=','))

    #only for page_2.txt
    f_path_2= path +'\page_2.txt'
    with open(f_path_2) as f:
        str_arr = ','.join([l.strip() for l in f])
    pgno_2 = np.asarray(str_arr.split(','), dtype=int)

    #using loop feach data from those text file.datda type = int
    for j in range(3,8):
    #store file path using variable
        txt_file_path=path+'\page_'+str(j)+'.txt'


        if os.path.exists(txt_file_path)==True:

            #genarate a variable name that auto incriment with for loop
            foo='pgno_'+str(j)
        else:
            break

        #pass the variable name as string and store value
        exec(foo + " = np.array(np.loadtxt(txt_file_path, dtype='i', delimiter=','))")

    #marge all array from page 2 to rest in single array in one dimensation
    f_array=np.concatenate((pgno_2,pgno_3,pgno_4,pgno_5,pgno_6,pgno_7), axis=0)

    #for first time of the loop assing this value
    if array_control_var==0:
        main_f_array=f_array
    if array_control_var==1:

        #here use np.array()
        main_f_array=np.array([main_f_array,f_array])
    else:
        main_f_array=np.insert(main_f_array, array_control_var, f_array, 0)

    array_control_var+=1

print(main_f_array)

我想要这样的输出

  

初始   [[0,0,0],[0,0,0,]]

     

插入后   [[0,0,0],[0,0,0],[0,0,0]]

但是输出是

[array([0,  0,  0])
array([0,  0,  0])
0 0 0]

2 个答案:

答案 0 :(得分:0)

正如我评论的那样,用insert(或concatenate的变体)收集数组很难做到正确,并且在工作时很慢。每次都会构建一个全新的数组。将数组收集到一个列表中,并在最后进行一个数组构建会变得更加轻松快捷。列表附加功能高效且易于使用。

也就是说,您报告的结果看起来可疑。我可以用以下方式重现它:

In [281]: arr = np.zeros(2, object)
In [282]: arr
Out[282]: array([0, 0], dtype=object)
In [283]: arr[0] = np.array([0,0,0])
In [284]: arr[1] = np.array([0,0,0])
In [285]: arr
Out[285]: array([array([0, 0, 0]), array([0, 0, 0])], dtype=object)
In [286]: np.insert(arr, 2, np.array([0,0,0]), 0)
Out[286]: array([array([0, 0, 0]), array([0, 0, 0]), 0, 0, 0], dtype=object)

在较早的迭代中,main_f_array必须已经创建为对象dtype数组。

如果它是一个“普通的”二维数组,则插入内容将有所不同:

In [287]: arr1 = np.zeros((2,3),int)
In [288]: np.insert(arr1, 2, np.array([0,0,0]), 0)
Out[288]: 
array([[0, 0, 0],
       [0, 0, 0],
       [0, 0, 0]])

或更像我想的那样迭代:

In [289]: f_array = np.array([0,0,0])
In [290]: main = f_array
In [291]: main = np.array([main, f_array])
In [292]: main
Out[292]: 
array([[0, 0, 0],
       [0, 0, 0]])
In [293]: main = np.insert(main, 2, f_array, 0)
In [294]: main
Out[294]: 
array([[0, 0, 0],
       [0, 0, 0],
       [0, 0, 0]])

答案 1 :(得分:0)

当我建议将insert替换为列表版本时,这就是我的想法。

import numpy as np

alist = []
for i in range(4): 
    f_array = np.array([i, i+2, i+4])
    alist.append(f_array)

print(alist)
main_f_array = np.array(alist)

print(main_f_array)

试运行:

1246:~/mypy$ python3 stack54715610.py 
[array([0, 2, 4]), array([1, 3, 5]), array([2, 4, 6]), array([3, 5, 7])]
[[0 2 4]
 [1 3 5]
 [2 4 6]
 [3 5 7]]

如果文件加载产生的数组大小不同,则会得到不同的结果

f_array = np.arange(i, i+1+i)

1246:~/mypy$ python3 stack54715610.py 
[array([0]), array([1, 2]), array([2, 3, 4]), array([3, 4, 5, 6])]
[array([0]) array([1, 2]) array([2, 3, 4]) array([3, 4, 5, 6])]

这是一个1d对象dtype数组,与2d相反。