Python教你将字符串转换成dataframe

Python教你将字符串转换成dataframe

在处理数据时，经常会遇到需要将字符串转换成Dataframe的情况。那么，该如何使用Python来实现呢？下面这段代码可以帮助你快速完成这一任务。

import pandas as pd
import io

# 将字符串转换成数据框
def str_to_dataframe(str_data):
    # 将字符串装换成字节流
    bytes_io = io.StringIO(str_data)
    
    # 使用pandas读取字节流并转换成Dataframe
    df = pd.read_csv(bytes_io, sep=',')
    
    return df

上述代码首先导入了pandas库，并使用io模块将字符串转换成字节流。接下来，使用pandas库中的read_csv函数读取字节流，并将其转换成Dataframe。最后返回生成的Dataframe。

使用上述代码，我们可以将一个逗号分隔的字符串转换成Dataframe，例如：

input_str = 'id,name,age\n001,Tom,23\n002,Bob,25\n003,Jerry,22\n'
df = str_to_dataframe(input_str)
print(df)

执行代码后，控制台将输出以下结果：

   id   name  age
0   1    Tom   23
1   2    Bob   25
2   3  Jerry   22

由此可知，编写一个函数将字符串转换成Dataframe非常简单，只需要使用字符串IO和pandas，就可以搞定！