Streamlit中Dataframe转TXT文件末尾出现空行的问题求助

问题出在Pandas的to_csv方法上——它默认会在输出内容的最后一行末尾添加一个换行符（\n），这是CSV格式的标准行为，但会导致生成的TXT文件末尾出现空行。你可以通过以下方式修改convert_df函数来解决：

方法1：使用rstrip('\n')去除末尾换行 #

直接对to_csv生成的字符串调用rstrip('\n')，只去掉末尾的换行符，不影响内容中的其他换行：

def convert_df(dataframe):
    csv_str = dataframe.to_csv(sep='\t', index=False, header=False).rstrip('\n')
    return csv_str.encode('utf-8')

方法2：切片去除最后一个字符 #

如果能确定to_csv的输出最后一定是换行符，可以直接用切片去掉最后一个字符：

def convert_df(dataframe):
    csv_str = dataframe.to_csv(sep='\t', index=False, header=False)[:-1]
    return csv_str.encode('utf-8')

方法3：兼容空Dataframe的安全处理 #

如果担心Dataframe为空时切片会报错，可以加个判断逻辑：

def convert_df(dataframe):
    csv_str = dataframe.to_csv(sep='\t', index=False, header=False)
    if csv_str and csv_str[-1] == '\n':
        csv_str = csv_str[:-1]
    return csv_str.encode('utf-8')

修改后重新运行代码，下载的TXT文件末尾就不会再出现多余的空行了。

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