python提取多个excel里想要的列

发布时间 2023-07-03 11:15:27作者: 蓝色大螃蟹
#!usr/bin/env python
# -*- coding:utf-8 -*-
"""
@author: Suyue
@file: calculatejsjp2.py
@time: 2023/06/30
@desc:
"""
#导入所需要的模块
import pandas as pd
import os

#定义文件路径
path1 = 'G:/drought/processdata/lianxi/everymonth/'
path2 = 'G:/drought/processdata/lianxi/month/'

#通过函数取出所有文件名称
files1 = os.listdir(path1)
files2 = os.listdir(path2)
def1 = pd.DataFrame()
def2 = pd.DataFrame()

#提取文件everymonth里的数据
for i in range(0,len(files1)):
    file_name = path1+files1[i]
    df = pd.read_excel(file_name)
    df = df.loc[:,['name','month','rain']].values
    df1 = pd.DataFrame(df)
    print(df1)


#提取文件month里的数据
for i in range(0,len(files2)):
    file_name = path2+files2[i]
    df2 = pd.read_excel(file_name)
    df2 = df2.loc[:,['name','month','rain']].values
    print(df2)