Pandas 模糊查詢與替換的操作

#!/usr/bin/python
# -*- encoding: utf-8 
import numpy as np
import pandas as pd
from fuzzywuzzy import fuzz
from fuzzywuzzy import process 
 
def enum_row(row):
 print(row['state']) #對state這一列做枚舉 
 
def find_state_code(row):
 if row['state'] != 0:  # 如果這個state的名字存在，就用state的名字與states列表中的值選擇一個最接近的，如果小於80分，直接舍棄，大於80才返回
 print(process.extractOne(row['state'], states, score_cutoff=80)) 
 
def capital(str): # 把str這個字符串，第一個字母大寫，其餘小寫
 return str.capitalize() 
 
def correct_state(row):
 if row['state'] != 0: # 如果這個state的名字存在，就用state的名字與states列表中的值選擇一個最接近的，如果小於80分，直接舍棄，大於80才返回
 state = process.extractOne(row['state'], states, score_cutoff=80)
 if state: # 如果為真，則找到瞭一個相關性的州名
  state_name = state[0] # 選擇用找到的這個州名數據
  return ' '.join(map(capital, state_name.split(' '))) # 先按空格分開（有的州名中間有空格）單詞，然後每個單詞首字母大寫
 return row['state'] 
 
def fill_state_code(row):
 if row['state'] != 0:
 state = process.extractOne(row['state'], states, score_cutoff=80)
 if state:
  state_name = state[0]
  return state_to_code[state_name] # 返回這個州名的value，即縮寫
 return '' 
 
if __name__ == "__main__":
 pd.set_option('display.width', 200) # 橫向最多顯示多少個字符， 一般80不適合橫向的屏幕，平時多用200.
 data = pd.read_excel('sales.xlsx', sheetname='sheet1', header=0) # 讀取excel表
 print('data.head() = \n', data.head()) # 默認顯示前五行
 print('data.tail() = \n', data.tail()) # tail顯示後五行
 print('data.dtypes = \n', data.dtypes) # 數據類型
 print('data.columns = \n', data.columns)# 顯示第一行行名
 for c in data.columns:
 print(c, end=' ') # 輸出第一行行名，中間以空格隔開
 print() #相當於回車
 data['total'] = data['Jan'] + data['Feb'] + data['Mar'] # Jan、Feb、Mar三列的值相加得到一個total
 print(data.head())
 print(data['Jan'].sum()) # Jan這一列的值相加
 print(data['Jan'].min()) # Jan這一列的最小值
 print(data['Jan'].max()) # Jan這一列的最大值
 print(data['Jan'].mean()) # Jan這一列的平均值
 
 print('=============')
 # 添加一行
 s1 = data[['Jan', 'Feb', 'Mar', 'total']].sum() # s1包含四個值，分別是這四列的和
 print(s1)
 s2 = pd.DataFrame(data=s1)
 print(s2)
 print(s2.T)
 print(s2.T.reindex(columns=data.columns)) # 將s2進行轉置輸出
 # 即：
 s = pd.DataFrame(data=data[['Jan', 'Feb', 'Mar', 'total']].sum()).T
 s = s.reindex(columns=data.columns, fill_value=0)
 print(s)
 data = data.append(s, ignore_index=True)
 data = data.rename(index={15:'Total'})
 print(data.tail())
 
 # apply的使用
 print('==============apply的使用==========')
 data.apply(enum_row, axis=1)  # axis=0時對每一列做變換，axis=1時對每一行做變換
 
 # 事先寫好以state為單位的編碼字典
 state_to_code = {"VERMONT": "VT", "GEORGIA": "GA", "IOWA": "IA", "Armed Forces Pacific": "AP", "GUAM": "GU",
   "KANSAS": "KS", "FLORIDA": "FL", "AMERICAN SAMOA": "AS", "NORTH CAROLINA": "NC", "HAWAII": "HI",
   "NEW YORK": "NY", "CALIFORNIA": "CA", "ALABAMA": "AL", "IDAHO": "ID",
   "FEDERATED STATES OF MICRONESIA": "FM",
   "Armed Forces Americas": "AA", "DELAWARE": "DE", "ALASKA": "AK", "ILLINOIS": "IL",
   "Armed Forces Africa": "AE", "SOUTH DAKOTA": "SD", "CONNECTICUT": "CT", "MONTANA": "MT",
   "MASSACHUSETTS": "MA",
   "PUERTO RICO": "PR", "Armed Forces Canada": "AE", "NEW HAMPSHIRE": "NH", "MARYLAND": "MD",
   "NEW MEXICO": "NM",
   "MISSISSIPPI": "MS", "TENNESSEE": "TN", "PALAU": "PW", "COLORADO": "CO",
   "Armed Forces Middle East": "AE",
   "NEW JERSEY": "NJ", "UTAH": "UT", "MICHIGAN": "MI", "WEST VIRGINIA": "WV", "WASHINGTON": "WA",
   "MINNESOTA": "MN", "OREGON": "OR", "VIRGINIA": "VA", "VIRGIN ISLANDS": "VI",
   "MARSHALL ISLANDS": "MH",
   "WYOMING": "WY", "OHIO": "OH", "SOUTH CAROLINA": "SC", "INDIANA": "IN", "NEVADA": "NV",
   "LOUISIANA": "LA",
   "NORTHERN MARIANA ISLANDS": "MP", "NEBRASKA": "NE", "ARIZONA": "AZ", "WISCONSIN": "WI",
   "NORTH DAKOTA": "ND",
   "Armed Forces Europe": "AE", "PENNSYLVANIA": "PA", "OKLAHOMA": "OK", "KENTUCKY": "KY",
   "RHODE ISLAND": "RI",
   "DISTRICT OF COLUMBIA": "DC", "ARKANSAS": "AR", "MISSOURI": "MO", "TEXAS": "TX", "MAINE": "ME"}
 states = list(state_to_code.keys()) # 把字典中的key拿出來放到states列表中
 print(fuzz.ratio('Python Package', 'PythonPackage')) #計算Python Package與PythonPackage的相似度
 print(process.extract('Mississippi', states)) # Mississippi與states中哪個最接近，並且列出相似比，不考慮大小寫
 print(process.extract('Mississipi', states, limit=1)) # limit=1代表隻取最接近的一個
 print(process.extractOne('Mississipi', states)) # extractOne代表隻取最接近的一個
 data.apply(find_state_code, axis=1) #apply表示對每一行（axis=1）的數據做find_state_code的變換
 
 print('Before Correct State:\n', data['state']) # 打印修改之前的state
 data['state'] = data.apply(correct_state, axis=1) # 檢測每一行，並對其修改
 print('After Correct State:\n', data['state'])
 data.insert(5, 'State Code', np.nan) # 插入State Code這一列，為這一列州名的縮寫
 data['State Code'] = data.apply(fill_state_code, axis=1)
 print(data)
 
 # group by
 print('==============group by================')
 print(data.groupby('State Code'))
 print('All Columns:\n')
 print(data.groupby('State Code').sum()) # 按州名縮寫劃分，並將同樣州名的數字相加
 print('Short Columns:\n')
 print(data[['State Code', 'Jan', 'Feb', 'Mar', 'total']].groupby('State Code').sum())
 
 # 寫入文件
 data.to_excel('sales_result.xls', sheet_name='Sheet1', index=False)