我正在从Matlab转向Python,所以对Python还很陌生。我试图创建一个基本的代码来进行一些数据分析。它应该读取指定目录中的所有.txt数据文件,并用实际.txt文件的名称进行标记。通过使用字典,我已经找到了一种方法使其正常工作,但如果有更好的方法,我将非常感激听到它。
一旦我可以访问数据,我想创建一个新列表,其中包含处理过的数据版本。为此,我想创建一个新的n x m列表或数组,但我找不到正确初始化这样一个列表的方法。我的最新尝试导致以下错误: 列表索引必须为整数,而不是元组。
以下是代码:
感谢任何反馈!
一旦我可以访问数据,我想创建一个新列表,其中包含处理过的数据版本。为此,我想创建一个新的n x m列表或数组,但我找不到正确初始化这样一个列表的方法。我的最新尝试导致以下错误: 列表索引必须为整数,而不是元组。
以下是代码:
import sys
import os
import re
import string
from numpy import *
listing = os.listdir(path)
dic = {} # define a dictionary to map the datafiles to while maintaining their filename
for filename in listing:
match = re.findall(r'[\w.]+\.txt', filename) # Use a regular expression findall function to identify all .txt files
if match:
dic[match.pop()[:-4]] = loadtxt(filename) # Drop the .txt and assign the datafile its original name
E = []
E[:,0] = dic['Test_Efield_100GHz'][:,0]
E[:,1] = dic['Test_Efield_100GHz'][:,1]
E[:,2] = abs(dic['Test_Efield_100GHz'][:,4]+dic['Test_Efield_100GHz'][:,7]*1j)**2
E[:,3] = abs(dic['Test_Efield_100GHz'][:,5]+dic['Test_Efield_100GHz'][:,8]*1j)**2
E[:,4] = abs(dic['Test_Efield_100GHz'][:,6]+dic['Test_Efield_100GHz'][:,9]*1j)**2
感谢任何反馈!