定义多维数组,并赋值为0,,根据多重嵌套的list长度利用len找出来,将最长的list作为数组在这一维的下标,然后再讲list值赋给多维数组,这样就将原来较短的list填充为0
import numpy as py
location=np.zeros([len(paths),len(max(paths,key=lambda x:len(x))),2])
ftype=np.zeros([len(smask),len(max(smask,key=lambda x:len(x))),1])for i,j in enumerate(paths):
location[i,0:len(j),0:2]=j
for i,j in enumerate(smask):
ftype[i,0:len(j),0:1]=j
其中enumerate()对一个列表,既要遍历索引又要遍历元素,多用于循环中
参考https://stackoverflow.com/questions/3685265/how-to-write-a-multidimensional-array-to-a-text-file