时间戳

首页 > 解决方案 > 如何从 read_csv 中选择以特定单词开头的特定列？

python-3.x - 如何从 read_csv 中选择以特定单词开头的特定列？

问题描述

我有一个很大的 csv 文件，将近 400 列我只想读取以“A”和“X”开头的列。例如，我有以下列：

当我使用 pandas 从 csv 文件中读取时，我只想选择：

而且我不知道我有多少个 As 和 X。

标签： python-3.xpandasdataframe

解决方案

您读取文件两次：一次仅用于标题，一次用于实际数据：

df = pd.read_csv('data.csv', usecols=lambda col: col.startswith('A_') or col.startswith('X_'))

推荐阅读