如何在python中基于数组的子集列表

Question

我在列表中有以下列名:

 vars = ['age','balance','day','duration','campaign','pdays','previous','job_admin.','job_blue-collar']

我有一个由数组索引组成的数组

(array([1, 5, 7], dtype=int64),)

我想基于数组索引对列表进行子集化

期望的输出应该是

vars = ['balance','pdays','job_admin.']

我在python中尝试过类似的东西

for i, a in enumerate(X):
   if i in new_L:
       print i

但是,它不起作用.

Answer 1

只需使用循环即可:

result=[]
for i in your_array:
   result.append(vars[i])

或一个线性

 [vars[i] for i in your_array]

Answer 2

如果您numpy无论如何都在使用，请使用其高级索引

import numpy as np
vars = ['age','balance','day','duration','campaign','pdays',
        'previous','job_admin.','job_blue-collar']
indices = (np.array([1, 5, 7]),)

sub_array = np.asarray(vars)[indices]  
# --> array(['balance', 'pdays', 'job_admin.'], dtype='<U15')

或者如果你想要一个list

sub_list = np.asarray(vars)[indices].tolist()
# --> ['balance', 'pdays', 'job_admin.']