4

我正在从 Matlab 转到 Python,所以我对 Python 还很陌生。我正在尝试为一些数据分析创建一段基本的代码。它应该读入指定目录中的所有 .txt 数据文件,并用实际 .txt 文件的名称标记它们。我已经设法找到一种方法来使用字典来完成这项工作,但如果有更好的方法,我将非常感激听到它。

一旦我可以访问数据,然后我想创建一个包含该数据的操纵版本的新列表。为此,我想创建一个新的 nxm 列表或数组,但是我找不到如何正确初始化这样的列表。我最近的努力导致以下错误:

列表索引必须是整数,而不是元组

代码如下:

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

感谢您的任何反馈!

4

2 回答 2

3

您在标准 Python类型 (your ) 变量上使用了仅numpy库支持的扩展切片语法。listE

您必须使用E.append()向它添加新值:

E.append(dic['Test_Efield_100GHz'][:,0])

或者简单地将整个列表定义为一组表达式:

E = [
    dic['Test_Efield_100GHz'][:,0]
    dic['Test_Efield_100GHz'][:,1]
    abs(dic['Test_Efield_100GHz'][:,4]+dic['Test_Efield_100GHz'][:,7]*1j)**2
    abs(dic['Test_Efield_100GHz'][:,5]+dic['Test_Efield_100GHz'][:,8]*1j)**2
    abs(dic['Test_Efield_100GHz'][:,6]+dic['Test_Efield_100GHz'][:,9]*1j)**2
]

或改用numpy数组。

标准的 python 序列类型,例如str,list并且tuple只支持简单的索引:

E[0]
E[-2]

或普通切片:

E[:10]
E[1:2]
E[3:]
E[:-1]
于 2012-11-15T17:10:14.400 回答
0

您只能为数组指定一个索引。
使用类似这样的东西
E.append(dic['Test_Efield_100GHz'][0])
这会将 E[0] 存储为一个数组,其值可以通过 E[0][i] 访问,其中 i 是列名
注意:Python 没有矩阵类型的数据结构,它只有数组。矩阵只是一个数组数组。

于 2012-11-15T17:13:48.030 回答