如何在bash中读取python的输入

时间:2016-08-19 09:45:59

标签: python bash

我有来自here的代码,我想在bash中使用这个python脚本,如:

  

python conversion.py sample.tsv sample.xlsx

python脚本:

import csv
from xlsxwriter.workbook import Workbook

# Add some command-line logic to read the file names.
tsv_file = 'sample.tsv'
xlsx_file = 'sample.xlsx'

# Create an XlsxWriter workbook object and add a worksheet.
workbook = Workbook(xlsx_file)
worksheet = workbook.add_worksheet()

# Create a TSV file reader.
tsv_reader = csv.reader(open(tsv_file, 'rb'), delimiter='\t')

# Read the row data from the TSV file and write it to the XLSX file.
for row, data in enumerate(tsv_reader):
    worksheet.write_row(row, 0, data)

# Close the XLSX file.
workbook.close()

如何修改此python脚本。对不起,这可能是一个愚蠢的问题,但我不使用python。

1 个答案:

答案 0 :(得分:2)

您可以使用sys.argv

...

tsv_file = sys.argv [1]

xlsx_file = sys.argv [2]

...

你可以像:

一样运行
main.py sample.tsv sample.xlsx

你也可以查看argparse包。它提供了许多奇特的功能。它基本上是sys.argv的包装器。