如何使用StringIO从内存以CSV格式发送数据帧

Question

我正在尝试通过电子邮件将熊猫数据帧作为CSV附件发送。

我的代码没有引发错误。但是，生成的电子邮件包含一个空附件。

我想念什么吗？

我正在使用mailgun api收发电子邮件，并尝试了以下代码：

from io import StringIO
import pandas as pd

def send_simple_message(a):
    mgurl = 'https://api.mailgun.net/v3/{}/messages'.format('sandboxf04fcce3c4dc46c987c92f3a967e7f9c.mailgun.org')
    auth = ('api', '3701ba6d2b1ad202e76a4322a80c7600-87cdd773-683e02b1')
    files=[("attachment", ("test.csv", a))]
    data = {
        'from': 'Mailgun User <mailgun@{}>'.format('sandboxf04fcce3c4dc46c654c92f3a765e7f9c.mailgun.org'),
        'to': 'user@gmail.com',
        'subject': 'Simple Mailgun Example',
        'text': 'Find attached your CSV'
    }

    response = requests.post(mgurl, auth=auth, data=data, files=files)
    response.raise_for_status()


a = StringI0()

df=pd.DataFrame({'a':[1,2,3],'b':[3,2,1]})

df.to_csv(a)

send_simple_message(a)

Answer 1

方法1：to_string()

a = StringIO(df.to_string(index=False))

send_simple_message(a)

方法2：

将其导出到csv，然后使用open函数进行读取，以便以str的形式返回：

df.to_csv('df.csv', index=False)

with open('df.csv') as f:
    lines = [line for line in f]

a = StringIO(' '.join(lines))

send_simple_message(a)

要检查我们的a变量是否已填充：

pd.read_csv(a)

   a  b
0  1  3
1  2  2
2  3  1