如何使用Pandas DataFrame返回逗号分隔格式的结果？

问题描述 #

当前使用以下代码生成并返回Pandas DataFrame：

arr1 = ['a','b','c']
arr2 = ['1','2','3']
data = { 'A': arr1, 'B': arr2 }
df = pd.DataFrame(data)
blank = [''] * len(arr1)
df.index = blank
return df

返回的DataFrame输出格式为：

A  B
a  1
b  2
c  3

需要从函数中返回逗号分隔的纯文本格式，而非DataFrame对象，预期结果如下：

A,B
a,1
b,2
c,3

询问是否可通过DataFrame实现该需求。

解决方案 #

完全可以实现，利用Pandas内置的to_csv()方法就能将DataFrame直接转为符合要求的CSV格式字符串，关键是配置参数去除多余的索引。

修改后的代码 #

import pandas as pd

def generate_csv_output():
    arr1 = ['a','b','c']
    arr2 = ['1','2','3']
    data = { 'A': arr1, 'B': arr2 }
    df = pd.DataFrame(data)
    # 转换为CSV字符串，禁用索引输出，保留表头
    csv_result = df.to_csv(index=False, header=True)
    # 可选：移除字符串末尾的多余换行符
    # csv_result = csv_result.rstrip('\n')
    return csv_result

参数说明 #

• index=False：原代码设置了空索引，此参数会直接跳过索引列的输出，避免CSV中出现多余的空列
• header=True：保留表头行（即"A,B"），如果不需要表头可将该值设为False
• 调用上述函数后，返回的就是你需要的逗号分隔格式文本

内容的提问来源于stack exchange，提问作者Vijay