Python之Pandas操作csv文件dataframe

 

# -*- coding: utf-8 -*-
# author：baoshan

import pandas as pd

def main():
    aqi_data = pd.read_csv('china_city_aqi_teacher.csv')
    print('基本信息:')
    print(aqi_data.info())
    print('数据预览')
    print(aqi_data.head(5))

    # 基本统计
    print('AQI最大值', aqi_data['AQI'].max())
    print('AQI最小值', aqi_data['AQI'].min())
    print('AQI均值', aqi_data['AQI'].mean())

    print(aqi_data[aqi_data.City=='聊城'])
    # top10
    bottom10_cities = aqi_data.sort_values(by=['AQI'], ascending=False).head(10)
    print('空气质量最差的十个城市')
    print(bottom10_cities)

    # 保存csv文件
    bottom10_cities.to_csv('bottom10_aqi.csv', index=False)


if __name__ == '__main__':
    main()

 

pandas操作dataframe示例，比csv模块写入csv简便了许多。