Python数据帧 - 如何选择所有列名中包含特定子字符串的列

12 浏览
0 Comments

Python数据帧 - 如何选择所有列名中包含特定子字符串的列

在Python中，我有一个数据框(df)，其中包含以下列名：A_OPEN，A_CLOSE，B_OPEN，B_CLOSE，C_OPEN，C_CLOSE，D_等等......\n如何轻松选择仅包含名称中包含_CLOSE的列？A，B，C，D，E，F等可以具有任何值，因此我不想使用特定的列名。\n在SQL中，可以使用like运算符来实现：df[like\'%_CLOSE%\']\n那么在Python中应该怎么做呢？

0

1 答案

0 Comments

Python数据框中，如何选择所有列名中包含特定子字符串的列？

问题的原因：在处理数据框时，有时候我们需要选择特定列名中包含特定子字符串的列。这个问题的出现是因为我们需要根据列名的模式来选择和操作数据。

解决方法：我们可以使用列表推导式来解决这个问题。下面是一个示例代码：

df = pd.DataFrame(
   columns = ['_CLOSE_A', '_CLOSE_B', 'C'],
   data = [[2,3,4], [3,4,5]]
)
print(df[[x for x in df.columns if "_CLOSE" in x]])

运行以上代码，输出结果如下：

   _CLOSE_A  _CLOSE_B
0         2         3
1         3         4

以上代码中，我们使用列表推导式来选择所有列名中包含"_CLOSE"子字符串的列，并将结果打印输出。

通过以上的解决方法，我们可以轻松选择和操作所有列名中包含特定子字符串的列，从而满足我们的需求。

0