pandas转为list去掉空格

pandas转为list去掉空格

在数据处理过程中，经常会使用到pandas库来处理和分析数据。有时候我们需要将pandas的DataFrame数据转换成列表，并且去掉空格。本文将详细介绍如何使用pandas将数据转为列表，并去掉空格。

1. 使用pandas将数据转为列表

首先，我们需要导入pandas库，并构造一个简单的DataFrame示例：

import pandas as pd

data = {'A': ['  apple', 'banana  ', '  orange '],
        'B': [1, 2, 3]}

df = pd.DataFrame(data)
print(df)

运行以上代码，可以得到DataFrame数据如下：

         A  B
0   apple  1
1  banana  2
2  orange  3

接下来，我们可以使用tolist()方法将DataFrame数据转为列表：

data_list = df.values.tolist()
print(data_list)

运行以上代码，可以得到转换后的列表如下：

[['  apple', 1], ['banana  ', 2], ['  orange ', 3]]

2. 去掉列表中的空格

在得到转换后的列表之后，下一步就是去掉列表中字符串元素的空格。我们可以使用列表推导式和strip()方法来实现这一功能：

cleaned_data_list = [[x.strip() if isinstance(x, str) else x for x in row] for row in data_list]
print(cleaned_data_list)

运行以上代码，可以得到去掉空格后的列表如下：

[['apple', 1], ['banana', 2], ['orange', 3]]

通过以上方法，我们成功将pandas的DataFrame数据转换成了列表，并且去掉了空格。这样就可以更方便地对数据进行后续处理和分析。

总结一下，本文详细介绍了如何使用pandas将数据转为列表，并去掉空格。