如何从 Pandas DataFrame 中提取数据作为列表？

从 Pandas DataFrame 中提取数据作为列表

在 Pandas DataFrame 中，每一列代表一个 Pandas Series，可以将其转换为 Python 列表。此功能可以实现数据的高效迭代，并允许进一步处理和可视化。

要将列的内容提取为列表，可以使用 .tolist() 方法或 list(x) 转换。例如，如果您有一个包含名为“cluster”的列的 DataFrame df，则可以使用以下代码以列表形式检索其内容：

column1_list = df['cluster'].tolist()

或者，如果您想获取整个列或row 作为列表，您可以使用 .to_numpy() 方法（将数据转换为 NumPy 数组）或 .values() 属性（返回表示 DataFrame 数据的 2D NumPy 数组）。

对于例如，要将整个第一列提取为列表，您可以使用：

column1_list = df[df.columns[0]].tolist()

或者，以数组形式检索第一行：

row1_array = df.iloc[0].to_numpy() # or df.iloc[0].values

通过利用这些技术，您可以轻松地将 Pandas DataFrame 中的数据提取到列表中以进行后续处理或分析。

以上是如何从 Pandas DataFrame 中提取数据作为列表？的详细内容。更多信息请关注PHP中文网其他相关文章！