import pandas as pd import numpy as np df = pd.read_csv("result.csv") # csv # df = pd.read_excel("sample.xlsx") # excel df.to_excel("sample.xlsx", index=False) # write data to excel print(df.head(10)) # include row title, read 0-3 row data, sum is 5 row, read 0-1 column data, sum is 2 column column = df.ix[:3, :2] column.to_excel("sample.xlsx", index=False)
def find_need_data(need_list): raw_data = pd.read_excel("raw_data.xlsx") all_need = []
need_list = ["vickey", "wu"] for n in need_list: # find need data need = raw_data[raw_data["filed"].str.contains(n)] all_need.append(need) # write all need data into excel pd.concat(all_need).to_excel("sample.xlsx", index=False)