Python for Data分析师。我正在解析具有数千个条目的Excel文档。到目前为止,我已经有了将Excel更改为csv的代码,将整个文档复制粘贴到文本文件“ |”中。明确地分开而不是昏迷。不幸的是,数据缺少信息,我只想复制和粘贴具有某些列条目的信息,列条目的范围是0到999。请帮助我编写一个代码,该代码将迭代/遍历行并搜索一个介于0到999之间的数字。如果找到,我要继续复制/粘贴该条目和该行中的其他3列。如果找不到,请告诉代码跳到下一行,而不是从该行的列中复制任何信息。这将被复制到文本文件中。
import pandas as pd
import os.path as op
def extract_export_columns(df, list_of_columns, file_path):
column_df = df[list_of_columns]
column_df.to_csv(file_path, index=False, sep="|")
#Orrginal file
input_base_path = 'C:somewhere'
main_df_data_file = pd.read_csv(op.join(input_base_path, 'somewhere '))
#Product of code
output_base_path = r'C:somewhere'
extract_export_columns(main_df_data_file,
['column1', 'column2', 'column3', 'etc'],
op.join(output_base_path, 'something.txt'))
extract_export_columns(main_df_data_file,
['column7'],
op.join(output_base_path, 'some_other_thing.txt'))