Python读取LTspice图导出

时间:2016-07-18 11:15:28

标签: python python-3.x pandas import pspice

我想用LT和matplotlib绘制来自LTspice的一些数据,我正在寻找一种解决方案,用Python从LTspice导入导出的绘图数据。

我发现使用Pandas无法做到这一点,因为数据的格式如下所示:

5.00000000000000e+006\t(2.84545891331278e+001dB,8.85405282381414e+001°)

是否有可能使用Pandas导入此内容(例如使用自己的方言)或有人知道一个简单的解决方法(比如逐行读取文件并提取值)?

更糟糕的是,在导出多个步骤的图表时,数据由

之类的行分隔
Step Information: L=410n  (Run: 2/4)

在Java中,我可能使用 Scanner 对象来读取数据。 Python中是否有类似的功能,甚至是将绘图数据导入Python的简单方法?

2 个答案:

答案 0 :(得分:1)

我不熟悉LTspice的导出绘图数据,所以我假设您提供的示例行的格式始终有效。

查看pandas-0.18文档(here)的IO Tools部分,我没有看到任何可用于数据格式的解析器实用程序。首先想到的是在填写pandas数据帧之前进行自己的解析和准备。

我假设您的问题的关键部分是解析数据文件,因为我玩过pandas和matplotlib已经有一段时间了,所以期待与之相关的错误。

实施例

这是一个快速的&脏 python3 脚本,用于将数据解析为字典列表,使用它构建pandas数据帧并使用DataFrame的plot方法绘制它。我试着解释评论中的步骤:

# ltspice.py
""" Use it as: 
    > python3 ltspice.py /path/to/datafile """

import pandas
import sys

data_header = "Time Gain Degree".split()

# Valid line example:
# 5.00000000000000e+006\t(2.84545891331278e+001dB,8.85405282381414e+001°) 

def parse_line(linestr):
    # ValueError and IndexError exceptions are used to mark the failure of
    # the parse.
    try:
        # First we split at the '\t' character. This will raise ValueError if
        # there is no \t character or there is more than 1 \t
        timestr, rest = linestr.split('\t')

        # Then we find the indexes of the '(' and ')' in the rest string.
        parenst, parenend = (rest.find('(')+1,  rest.find(')'))
        if (parenst == -1) or (parenend == -1):
            # find() method returns -1 if nothing is found, I raise ValueError
            # to mark it as a parsing failure
            raise ValueError

        # rest[parenst:parenend] returns the string inside parens. split method
        # splits the string into words separated by the given character (i.e.
        # ',')
        powstr, degstr = rest[parenst:parenend].split(',')

        # converting strings into floats. Replacing units as necessary.
        time = float(timestr)
        power = float(powstr.replace('dB', ''))

        # this will fail with python 2.x
        deg = float(degstr.replace('°', ''))

        # You can use dict() instead of tuple()
        return tuple(zip(data_header, (time, power, deg)))

    except (ValueError,IndexError) as e:
        return None


def fileparser(fname):
    """ A generator function to return a parsed line on each iteration """
    with open(fname, mode='r') as fin:
        for line in fin:
            res = parse_line(line)
            if res is not None:
                yield res

def create_dataframe(fname):
    p = fileparser(fname)
    # rec is a tuple of 2-tuples that can be used to directly build a python
    # dictionary
    recs = [dict(rec) for rec in p]
    return pandas.DataFrame.from_records(recs)

if __name__ == '__main__':
    data_fname = sys.argv[1]
    df = create_dataframe(data_fname)

    ax = df.plot(x='Time', y='Gain')
    fig = ax.get_figure()
    fig.savefig('df.png')

您可以将此代码复制到文本编辑器并将其另存为ltspice.py,然后在终端中使用python3 ltspice.py yourdata.dat运行。

请注意,parse_line函数实际上以('key',value)的形式返回2元组的元组,其中'key'表示列名。然后,此值用于构建create_dataframe函数中的字典列表。

附加

我写了另一个脚本来测试行为:

# test.py
import random
from ltspice import fileparser


def gen_data():
    time = random.randint(0,100)*1e6
    db = random.lognormvariate(2,0.5)
    degree = random.uniform(0,360)
    # this is necessary for comparing parsed values with values generated
    truncate = lambda x: float('{:.15e}'.format(x))
    return (truncate(time),truncate(db),truncate(degree))


def format_data_line(datatpl):
    time, db, degree = datatpl[0], datatpl[1], datatpl[2]
    formatted = "{0:.15e}\t({1:.15e}dB,{2:.15e}°)\n"
    return formatted.format(time, db, degree)


def gen_ignore_line():
    tmpl = "Step Information: L={}n  (Run:{}/{})\n"
    l = random.randint(100,1000)
    r2 = random.randint(1,100)
    r1 = random.randint(0,r2)
    return tmpl.format(l,r1,r2)


def create_test_file(fname, valid_count, invalid_count):
    """ Creates a test file containing data lines mixed with lines to be
    ignored. Returns the data created.
    valid_count: number of the actual data lines
    invalid_count: number of the to-be-ignored lines
    """ 
    header = 'Time Gain Degree'.split()
    data = []
    formatteddatalines = []
    for i in range(valid_count):
        unfmtdata = gen_data()
        data.append(tuple(zip(header, unfmtdata)))
        formatteddatalines.append(format_data_line(unfmtdata))

    invalidlines = []
    for i in range(invalid_count):
        invalidlines.append(gen_ignore_line())

    lines = formatteddatalines + invalidlines
    random.shuffle(lines)
    with open(fname, mode='w') as fout:
        fout.writelines(lines)

    return data

if __name__ == '__main__':
    fname = 'test.data'
    validcnt = 10
    invalidcnt = 2

    validdata = create_test_file(fname, validcnt, invalidcnt)
    parseddata = [data for data in fileparser(fname)]

    # Note: this check ignores duplicates.
    assert(set(validdata) == set(parseddata))

答案 1 :(得分:1)

我为LTSpice模拟输出文件编写了一个Python阅读器,您可以在这里找到它:LTSPy。此处还有一些关于如何使用阅读器的示例文件:exltspy.zip。希望有用。 (我提前为我的草率编码做法道歉。)