一种在另一个序列中查找一个序列的所有索引的pythonic方法？ [复制]

Question

给定两个列表 f=['a','b','c','d'] 和 g=['a','c']，我可以生成 [0,2]=indexes(f,g)

def indexes(f,g):
    return [f.index(x) for x in g]

有没有更好的方法？

动机： 对于numpy矩阵X，我想提取一些列，所以我想用X[:,indexes(f,g)]替换pd.DataFrame(X,columns=f)[g].values。

Answer 1

您可以在np.1d 之后使用np.where 来获取相交索引：

np.where(np.in1d(f,g))
(array([0, 2], dtype=int64),)