在另一个序列中查找一个序列的所有索引的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),)