使用熊猫从Excel文件读取特定列

时间:2019-05-23 06:51:06

标签: python excel pandas

我有一个文件夹,里面装有所有excel文件和子文件夹。子文件夹也包含excel文件。

使用os.walk()我已经能够遍历每个子文件夹并从excel文件中获取数据。我的问题是,我只想从所有excel文件中获取手机号码列并将其存储在数据库中。问题在于,每个excel文件中手机号码的列名都不同,并且有数千个excel文件。

所以我决定按其数据类型和长度来获取列。 (例如,如果列数据类型为整数,并且该特定列中每个单元格的长度为10,则我会推送该数据。)

但是我不知道该怎么做。我遍历了列和行,但是得到的只是重复数据或错误。有人可以帮我这个忙吗?如果您能帮助我,因为两天以来我一直在这样做,但失败了,我们将非常高兴。预先感谢。

 def file_access(file_path_list):
    for path in file_path_list:
        if path.endswith(('xlsx' , 'xls' , 'XLS')):
            print '------------------------------\n\n'
            folder = path.split('/')[-2]
            sheet = path.split('/')[-1]
            print folder , sheet

            df = pd.read_excel(path, sheet_name=None, header=None)

            # Here i am trying to get data from df object but failing to do so.
            # You can suggest me code from this stage. 

1 个答案:

答案 0 :(得分:0)

要获取许多excel文件中的电话号码,您可以尝试以下代码:

import pandas as pd
import xlrd
import os

mydir = (os.getcwd()).replace('\\','/') + '/'

#Get all excel files include subdir
filelist=[]
for path, subdirs, files in os.walk(mydir):
    for file in files:
        if (file.endswith('.xlsx') or file.endswith('.xls') or file.endswith('.XLS')):
            filelist.append(os.path.join(path, file))
number_of_files=len(filelist)

# Get data of cells from excel
data=[]
for i in range(number_of_files):
    #df.append(pd.read_excel(r''+ mydir +filelist[i]))
    df=pd.read_excel(r''+filelist[i])
    l=len(df.iloc[0])
    for n in range(l):
        if len(str(df.iloc[0][n])) >= 10:
            data.append(df.iloc[:][df.axes[1][n]])
            break
res=[]
for i in range(len(data)):
    res.append(data[i].values.tolist())
print(res)

要从不同列的名称中获取所有电话数据,它使用电话号码的长度来区分其他列数据。在这种情况下,我使用的是我的国家/地区的电话号码,长度为11(例如:82330403045)。

输出:

>>> data
[0    82330403045
1    82330403046
2    82330403047
3    82330403048
Name: Phone, dtype: int64, 0    82330403049
1    82330403050
2    82330403051
3    82330403052
Name: PhoneCell, dtype: int64]

>>> res
[[82330403045, 82330403046, 82330403047, 82330403048], [82330403049, 82330403050, 82330403051, 82330403052], [82330403049, 82330403050, 82330403051, 82330403052], [82330403045, 82330403046, 82330403047, 82330403048], [82330403049, 82330403050, 82330403051, 82330403052], [82330403049, 82330403050, 82330403051, 82330403052]]

您可以将此输出数据用于数据库。

我的Excel文件:

Book1.xlsx文件:

Book2.xlsx文件: