在另一个序列中查找一个序列的所有索引的pythonic方法？

Question

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

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

有没有更好的办法？

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

Answer 1

您可以使用np.whereafter np.1d来获取交叉索引：np.where(np.in1d(f,g)) (array([0, 2], dtype=int64),)