读取代码

由于赛题给定的数据集文件都比较大,文件行数都比较多,如果完全进行读取可会让电脑卡死,内存爆炸;

  • 可以只读取部分文件,读取单个文件完成数分析;
  • 修改字段类型节约空间;

接下来看单个文件和部分文件的读取方法

1. 巡游车GPS数据读取
在读取数据时,可以完成以下操作:

  • 通过read_csv函数输入路径和nrows设置读取的文件行数;
  • nrows可以取值为数值或者None,前者数值控制行数,后者读取所有(默认是读取所有);
  • 读取完成后的数据为DataFrame形式,可以通过describe()函数完成描述型分析统计;

建议大家先设置nrows完成读取,不要直接读取所有文件;

import pandas as pd
import numpy as np# 文件目录,相对路径
INPUT_PATH = 'E:/python-project/deep-learning/datawhale/2020DCIC/data/taxiGps20190531/'# 文件读取行数
MAX_ROWS = 100000 taxigps2019 = pd.read_csv(INPUT_PATH + 'taxiGps20190531.csv', nrows=MAX_ROWS)
taxigps2019.describe()

DCIC巡游车与网约车运营特征对比分析-数据读取-编程知识网

  • 通过info()函数完成表格信息展示;
taxigps2019.info()

DCIC巡游车与网约车运营特征对比分析-数据读取-编程知识网
通过describe()info()函数,我们可以初步得到现有的字段的取值,10w条数据占用5MB内存。我们也可以完成数据字段压缩的操作,将字段类型根据取值空间进行修改,压缩内存使用需求。

import pandas as pd
import numpy as npINPUT_PATH = 'E:/python-project/deep-learning/datawhale/2020DCIC/data/taxiGps20190531/' #文件目录
MAX_ROWS = 100000 # 文件读取行数taxigps2019 = pd.read_csv(INPUT_PATH + 'taxiGps20190531.csv', nrows=MAX_ROWS,dtype = {'DRIVING_DIRECTION': np.uint16,'OPERATING_STATUS': np.uint8,'LONGITUDE': np.float32,'LATITUDE': np.float32,'GPS_SPEED': np.float16 })taxigps2019.info()

DCIC巡游车与网约车运营特征对比分析-数据读取-编程知识网
为了方便查看数据,我们还可以对GPS数据进行排序,这样就完成了单个文件的读取。

taxigps2019 = taxigps2019[taxigps2019.columns[::-1]]
taxigps2019.sort_values(by=['CARNO','GPS_TIME'], inplace=True)
taxigps2019.reset_index(inplace=True, drop=True)
taxigps2019.head()

DCIC巡游车与网约车运营特征对比分析-数据读取-编程知识网
多个文件(多天)可以直接将文件进行拼接即可:

# 出租车2019年GPS
taxigps2019 = pd.concat([pd.read_csv(INPUT_PATH + 'taxiGps20190531.csv', nrows=MAX_ROWS,dtype = {'DRIVING_DIRECTION': np.uint16,'OPERATING_STATUS': np.uint8,'LONGITUDE': np.float32,'LATITUDE': np.float32,'GPS_SPEED': np.float16 }),pd.read_csv(INPUT_PATH + 'taxiGps20190601.csv', nrows=MAX_ROWS,dtype = {'DRIVING_DIRECTION': np.uint16,'OPERATING_STATUS': np.uint8,'LONGITUDE': np.float32,'LATITUDE': np.float32,'GPS_SPEED': np.float16 })
])
taxigps2019 = taxigps2019[taxigps2019.columns[::-1]]
taxigps2019['GPS_TIME'] = pd.to_datetime(taxigps2019['GPS_TIME'])
taxigps2019.sort_values(by=['CARNO','GPS_TIME'], inplace=True)
taxigps2019.reset_index(inplace=True)

2. 巡游车订单读取
巡游车订单数据单个文件读取:

taxiorder2019 = pd.read_csv(INPUT_PATH + 'taxiOrder20190531.csv', nrows=MAX_ROWS,dtype = {'GETON_LONGITUDE': np.float32,'GETON_LATITUDE': np.float32,'GETOFF_LONGITUDE': np.float32,'GETOFF_LATITUDE': np.float32,'PASS_MILE': np.float16,'NOPASS_MILE': np.float16,'WAITING_TIME': np.float16})taxiorder2019 = taxiorder2019.rename(columns={'CAR_NO':'CARNO'})
taxiorder2019.sort_values(by=['CARNO','GETON_DATE'], inplace=True)
taxiorder2019.reset_index(inplace=True, drop=True)

巡游车订单数据多个文件读取:

taxiorder2019 = pd.concat([pd.read_csv(INPUT_PATH + 'taxiOrder20190531.csv', nrows=MAX_ROWS,dtype = {'GETON_LONGITUDE': np.float32,'GETON_LATITUDE': np.float32,'GETOFF_LONGITUDE': np.float32,'GETOFF_LATITUDE': np.float32,'PASS_MILE': np.float16,'NOPASS_MILE': np.float16,'WAITING_TIME': np.float16}),pd.read_csv(INPUT_PATH + 'taxiOrder20190601.csv', nrows=MAX_ROWS,dtype = {'GETON_LONGITUDE': np.float32,'GETON_LATITUDE': np.float32,'GETOFF_LONGITUDE': np.float32,'GETOFF_LATITUDE': np.float32,'PASS_MILE': np.float16,'NOPASS_MILE': np.float16,'WAITING_TIME': np.float16})
])
taxiorder2019 = taxiorder2019.rename(columns={'CAR_NO':'CARNO'})
taxiorder2019.sort_values(by=['CARNO','GETON_DATE'], inplace=True)
taxiorder2019.reset_index(inplace=True, drop=True)

3. 网约车GPS数据读取

wycgps2019 = pd.read_csv(INPUT_PATH + 'wycGps20190531.csv', nrows=MAX_ROWS,dtype={'LONGITUDE': np.float32,'LATITUDE': np.float32,'SPEED': np.float16})wycgps2019 = wycgps2019.rename(columns={'CAR_NO':'CARNO'})
wycgps2019 = wycgps2019[wycgps2019.columns[::-1]]
wycgps2019.sort_values(by=['CARNO','POSITION_TIME'], inplace=True)wycgps2019['BIZ_STATUS'] = wycgps2019['BIZ_STATUS'].fillna(-1).astype(np.int8)
wycgps2019['ENCRYPT'] = wycgps2019['ENCRYPT'].fillna(-1).astype(np.int8)

4. 网约车订单数据读取

wycorder2019 = pd.read_csv(INPUT_PATH + 'wycOrder20190531.csv', nrows=MAX_ROWS,dtype={'DEP_LONGITUDE': np.float32,'DEP_LATITUDE': np.float32,'DEST_LONGITUDE': np.float32,'DEST_LATITUDE': np.float32,})
wycorder2019 = wycorder2019.rename(columns={'CAR_NO':'CARNO'})
wycorder2019.sort_values(by=['CARNO','DEP_TIME'], inplace=True)

数据统计

赛题数据基本可以分为四类,不同类型的赛题数据在字段格式上有一定差异:

  • 巡游车GPS数据(2019年、2020年);
  • 巡游车订单数据(2019年、2020年);
  • 网约车GPS数据(2019年、2020年);
  • 网约车订单数据(2019年、2020年);

为了方便大家学习,接下来我们将以巡游车GPStaxiGps20190531.csv为案例进行数据统计:

  • 有多少辆出租车:
    • taxigps2019[‘CARNO’].nunique()
  • 出租车平均GPS速度:
    • np.clip(taxigps2019[‘GPS_SPEED’].values, 0, 150).mean()
  • 出租车运营状态统计:
    • taxigps2019[‘OPERATING_STATUS’].value_counts()
  • 某辆巡游车数据:
    • taxigps2019[taxigps2019[‘CARNO’] == ‘0006d282be70d06881a7513b69fcaa60’]
  • 某个运行方向的车辆统计:
    • taxigps2019[taxigps2019[‘DRIVING_DIRECTION’] == 10][‘CARNO’].unique()
  • 统计记录最多的GPS小时:
    • taxigps2019[‘GPS_TIME’] = pd.to_datetime(taxigps2019[‘GPS_TIME’])
    • taxigps2019[‘GPS_TIME’].dt.hour.value_counts()