Python 列表数据输出集成：txt|xlsx|csv
代码实现思路
为了更灵活地输出数据，我们创建了一个数据打印工具，该工具可以将二维列表的数据以不同的格式输出，包括文本文件(txt)、Excel文件(xlsx)和CSV文件(csv)。我们使用Python中的Pandas库来处理数据，并提供了一个简单的命令行界面示例。
函数要点整理
print_data 函数
这个函数根据用户选择的输出格式将数据打印到相应的文件中。具体要点包括：
-
输入参数:
data: 要打印的数据，以二维列表形式提供。output_format: 输出格式的选择，0表示txt，1表示Excel，2表示CSV。
-
输出结果:
- 根据选择的格式将数据输出到相应文件。
整体代码（附加注释）
import pandas as pd
from os import getcwd
def print_data(data, output_format):
"""
根据指定的输出格式打印数据。
:param data: 要打印的数据，二维列表形式
:param output_format: 输出格式选择，0 为 txt，1 为 Excel，2 为 CSV
"""
if output_format == 0:
with open('output.txt', 'w') as f:
for row in data:
f.write('\t'.join(map(str, row)) + '\n')
elif output_format == 1:
df = pd.DataFrame(data)
df.to_excel('output.xlsx', index=False, header=False)
elif output_format == 2:
df = pd.DataFrame(data)
df.to_csv('output.csv', index=False, header=False)
else:
print("Invalid output format choice. Choose 0 for txt, 1 for Excel, or 2 for CSV.")
def main():
# 示例多维列表
sample_data = [
['中南财经政法大学南湖校区', '114.385177,30.474255', '027-88386114', '湖北省', '武汉市', '江夏区',
'东湖新技术开发区南湖大道182号'],
['湖北工业大学', '114.308919,30.478607', '027-59751111', '湖北省', '武汉市', '洪山区',
'南李路28号(湖工大地铁站B口步行300米)'],
['武汉工程大学流芳校区', '114.432657,30.461083', '027-87905138', '湖北省', '武汉市', '江夏区',
'东湖新技术开发区光谷一路206号'],
['武汉老年大学', '114.287970,30.638652', '027-82774421', '湖北省', '武汉市', '江岸区', '金桥大道26号'],
['武汉理工大学南湖校区', '114.332921,30.508522', '027-87859017', '湖北省', '武汉市', '洪山区', '雄楚大道']
]
# 选择输出格式，0 为 txt，1 为 Excel，2 为 CSV
output_format_choice = 1
# 执行打印
print_data(sample_data, output_format_choice)
# 打印当前文件夹路径
print('文件输出路径为:', getcwd())
if __name__ == "__main__":
main()
注意事项
- 请确保Pandas库已经安装，可以使用
pip install pandas进行安装。 - 在运行程序之前，根据需要修改示例数据和输出路径。
术语与函数解释
| 术语/函数 | 解释 |
|---|---|
pd.DataFrame |
Pandas库中的数据结构，用于存储和处理二维表格数据。 |
to_excel |
Pandas DataFrame对象的方法，用于将数据写入Excel文件。 |
to_csv |
Pandas DataFrame对象的方法，用于将数据写入CSV文件。 |
getcwd |
os库中的方法，返回当前工作目录路径。 |
如果这对您有所帮助，希望点赞支持一下作者！
点击查看原文
