常用Python脚本

一、对文件下下所有图片进行编号

#coding=utf-8
import os

class ImageRename():
    def __init__(self):
        self.path = '/home/gavyn/桌面/data2/left/right'

    def rename(self):
        filelist = os.listdir(self.path)
        total_num = len(filelist)

        i = 0
        for item in filelist:
            if item.endswith('.png'):
                src = os.path.join(os.path.abspath(self.path), item)
                dst = os.path.join(os.path.abspath(self.path), 'right' + '0000' + format(str(i), '0>3s') + '.png')
                os.rename(src, dst)
                print 'converting %s to %s ...' % (src, dst)
                i = i + 1
        print 'total %d to rename & converted %d png' % (total_num, i)

if __name__ == '__main__':
    newname = ImageRename()
    newname.rename()

二、删除文本中空格

# coding: utf-8
import os
import sys

lines = open('/home/gavyn/ORBSLAM/ORB_SLAM2/Examples/Stereo/EuRoC_TimeStamps/outdoor.txt').readlines() #打开文件，读入每一行
print lines
fp = open('/home/gavyn/ORBSLAM/ORB_SLAM2/Examples/Stereo/EuRoC_TimeStamps/outdoor_1.txt','w') #打开你要写得文件pp2.txt
for s in lines:
    fp.write(s.replace(' ','')) # replace是替换，write是写入
fp.close() # 关闭文件
print 'ok'

三、对csv指定列进行处理

#coding=utf-8

import csv

with open('imu0.csv', 'r') as f:
    first_line = f.readline().strip()
    first = first_line.split(",")
    print(first)
    lines = f.readlines()

with open('imu.csv', 'w') as file:
    writer = csv.writer(file)
    # 先写入columns_name
    writer.writerow(first)

    #writer.writerows(num)
    for line in lines:
        num = line.split(",")
        num[4] = float(num[4]) * 9.8
        num[5] = float(num[5]) * 9.8
        num[6] = float(num[6]) * 9.8
        writer.writerow(num)