python - 基于列名的子集结构化数组

Question

我有一个从 CSV 文件创建的结构化数组。

我有很多字段/列。我想创建一个子集数组。

z=mydata[['z1','z2','z3']] 可以解决问题，但有时 z 来自 z1-仅 z2 或 z1-z10。

是否有一种简单的方法可以对列名以 z 开头的所有列进行子集化，无论数据中有多少列？显然，生成的数组将具有不同的列数，但这没关系。

Answer 1

不太漂亮，但您可以执行以下操作:

z = mydata[[x for x in a.dtype.names if 'z' in x]]

实际上，您循环遍历所有列名称并检查列名称是否符合条件。如果有更好的方法，我会非常感兴趣，因为我使用 pd.DataFrame.columns 属性在 pandas 中执行类似的操作。