获取当前目录下所有文件

#!/usr/bin/env python3a
# -*- coding: utf-8 -*-
"""
Created on Sun Dec 15 15:37:40 2019

@author: lg
"""

import os
import pandas as pd
from pymongo import MongoClient

client=MongoClient('localhost',27017)

db=client.glory.snapshot


def file_name(file_dir):
#    v=[]
    for root, dirs, files in os.walk(file_dir):
#        print('root_dir:', root)  # 当前目录路径
#        print('sub_dirs:', dirs)  # 当前路径下所有子目录
        print('files:', files)  # 当前路径下所有非目录子文件
        return files

c=file_name('./')


def save_data(df):
    num=df.shape[0]
    print('num=',num)
    
    for q in range(num):
        
        if q %10000==0:
            print(round(q/num,3))
        m=df.loc[q]
        
        m1=dict(m)
        
        if  str( m1['giftbag_id']) not  in ['nan','None'] :
            m1['giftbag_id']=int(m1['giftbag_id'])
#            
        m1['player_id']=int(m1['player_id'])
        
        db.save(m1)
#        db.replace_one({'player_id':m1['player_id'],'timestamp':m1['timestamp']},
#    
#                     m1,True
#                 )

kk=[]
for m in c:
    dt=pd.read_csv(m)
    ll=[ 'player_id', 'giftbag_id', 'status', 'timestamp', 'operation']
    
    dt1=dt[ll]
    save_data(dt1)
    
    
#    kk.append(dt)
#    print(m)
#
#m=pd.concat(kk)
#m.to_csv('eco.csv',index=False)
import os
import pandas as pd

def file_name(file_dir):
#    v=[]
    for root, dirs, files in os.walk(file_dir):
#        print('root_dir:', root)  # 当前目录路径
#        print('sub_dirs:', dirs)  # 当前路径下所有子目录
        print('files:', files)  # 当前路径下所有非目录子文件
        return files

c=file_name('./')


kk=[]
for m in c:
    dt=pd.read_csv(m)
    kk.append(dt)
    print(m)

m=pd.concat(kk)
m.to_csv('eco.csv',index=False)

posted @   luoganttcc  阅读(4)  评论(0编辑  收藏  举报
相关博文:
阅读排行:
· TypeScript + Deepseek 打造卜卦网站:技术与玄学的结合
· 阿里巴巴 QwQ-32B真的超越了 DeepSeek R-1吗?
· 【译】Visual Studio 中新的强大生产力特性
· 10年+ .NET Coder 心语 ── 封装的思维:从隐藏、稳定开始理解其本质意义
· 【设计模式】告别冗长if-else语句:使用策略模式优化代码结构
点击右上角即可分享
微信分享提示