如何在指定列数的情况下将字符串转换为数据帧？

pip install pandas

import pandas as pd

# 假设我们有一个长字符串，它包含了多行数据，每行的列数相同
data_string = """
1,John,Doe,30
2,Jane,Smith,25
3,Mike,Tyson,45
"""

# 指定每行的列数
num_columns = 4

# 使用splitlines()方法按行分割字符串，然后对每一行使用split()方法按逗号分割
rows = data_string.strip().splitlines()
data = [row.split(',') for row in rows]

# 创建数据帧
df = pd.DataFrame(data, columns=[f'Column{i+1}' for i in range(num_columns)])

# 显示数据帧
print(df)

   Column1 Column2 Column3 Column4
0        1    John     Doe      30
1        2    Jane    Smith      25
2        3    Mike    Tyson      45

# 检查每行是否有正确数量的列
for row in data:
    if len(row) != num_columns:
        raise ValueError(f"Row has an incorrect number of columns: {row}")