Python中标准库argparse的使用

 

import argparse
import pandas as pd


def data_format(fin):
	df = pd.read_excel(fin)
	newcolumns = ["peptides"]+[i.split("Intensity")[1].strip() for i in df.columns if i.startswith("Intensity")]
	data = []
	for row in df.itertuples():
		lines = list(row)
		if pd.isnull(lines[1]):
			continue
		pros = lines[1].split(";")
		pos = str(lines[2]).split(";")
		maps = zip(pros,pos)
		for mapv in maps:
			data.append(["-".join(list(mapv)).split("CON__")[-1]]+lines[3:])
	out = pd.DataFrame(data)
	out.columns = newcolumns
	out.to_excel(fin+"_format.xlsx",index=False)


def main():
	parser = argparse.ArgumentParser(description="formatdata")
	parser.add_argument("-i","--input")
	args = parser.parse_args()
	print(args.input)
	data_format(args.input)



if __name__ == "__main__":
	main()

　　

parser = argparse.ArgumentParser(description="formatdata")
parser.add_argument("-i","--input")
args = parser.parse_args()
print(args.input)
data_format(args.input)

 

if __name__ == "__main__":
main()