如何将一堆混合元素类型附加到另一个数组?

问题描述 投票:0回答:1

我有一个numpy数组,我正在使用if循环来过滤掉一些不需要的条目,并且我想将要保留的条目追加到新的“过滤后的数组”。

from astropy.table import Table
import numpy as np

data = '/home/myname/data.fits'
data = Table.read(data, format="fits") 

ID = np.array(data['id'])
ID.astype(str)

redshift = np.array(data['z'])
redshift.astype(float)

radius = np.array(data['r']) 
radius.astype(float)

mag = np.array(data['magnitude'])
mag.astype(float)

def stack(array1, array2, array3, array4):
    #stacks multiple arrays to have corresponding values next to eachother
    stacked_array = [(array1[i], array2[i], array3[i], array4[i]) for i in range(0, array1.size)]
    stacked_array = np.array(stacked_array)
    return(stacked_array) 

stacked = stack(ID, redshift, radius, mag)

filtered_array = np.array([])

for i in stacked:
    if i.any == 'nan':
        np.delete(stacked[i])
    else:
       np.append(filtered_array, stacked[i]) 

运行此文件时出现以下错误:

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

是否有任何方法可以在不出现此错误的情况下追加到新数组?

python arrays numpy append
1个回答
0
投票

我真是一团糟,我只需要从以下位置更改最后一行:

np.append(filtered_array, stacked[i]) 

to

np.append(filtered_array, i)
© www.soinside.com 2019 - 2024. All rights reserved.