python_將客流od數據轉成csv矩陣形式

本文章數據來源請參考這篇文章：
python_pandas_將街道行政區級別的數據進行分組求和

import pandas as pd
from utils.read_write import readCsv, writeCSV

'''
district_O,district_D,od
1區,2區,2919139
'''
def district():
    file = 'od_district_004.csv'
    data = readCsv(dir+file)
    fill = []
    finish = ['行政區']
    for i in range(1,len(data)):
        if i <14:
            finish.append(data[i][1])
    fill.append(finish)
    data2 = pd.read_csv(dir + file)
    groups = data2.groupby('district_O')
    for name,group in groups:
        finish = []
        finish.append(name)
        for one in group.itertuples(index=False):
            finish.append(one[2])
        fill.append(finish)
    writeCSV(fill, dir + '行政區客流矩陣' + "_" + '004' + '.csv')

if __name__ == '__main__':
    dir = r'D:\學習文件\數據庫\201\\'
    streetFile = '街道169.txt'
    district()
    file = 'od_street_04.csv'
    street = readCsv(dir+streetFile)
    data = readCsv(dir+file)
    fill = []
    title = ['街道']
    for i in range(0,len(street)):
        title.append(street[i][0])
    fill.append(title)
    data2 = pd.read_csv(dir + file)
    groups = data2.groupby('street_O')
    for name,group in groups:
        finish = []
        names = []
        finish.append(name)
        for one in group.itertuples(index=False):
            d = one[1]
            names.append(d)
        combine = []
        for name in names:
            for k in range(1, len(title)):
                if title[k] == name:
                    combine.append(title.index(name))
        for m in range(1, 169):
            if m in combine:
                d_name = names[combine.index(m)]
                col_value = group[group['street_D']==d_name].iat[0,2]
                finish.append(col_value)
            else:
                finish.append(0)
        fill.append(finish)
    writeCSV(fill, dir + '街道客流矩陣' + "_" + '24' + '.csv')

其中的參考文件請移步到我的下載
我的下載

如需數據或者幫忙處理數據請私聊我.。。
如果覺得寫得好請關注我哦，原創不易