切片 pandas 数据框以获取不连续的列

Question

我有一个pandas.DataFrame: wordvecs_df，列按顺序标记为'word'、'count'、'v1'through'v50'和'norm1'through 。我想创建一个新的 pandas df，其中仅包含,和 的'norm50'列。'word''count'norm1-norm50

wordvecs_df.loc[:,"norm1":"norm50"]

让我norm1- norm50，但如果我尝试输入单词并计数，我会得到一个 IndexingError: Too much Indexers。

我不知道如何从数据框中获取我想要的列。有任何想法吗？

Answer 1

您可以建立一个列名称列表，例如：

columns = ['word', 'count'] + ['norm%d' % i for i in range(1, 51)]
wordvecs_df.loc[:,columns]