python解析mdf或mf4文件利器之asammdf用法
作者:羸弱的穷酸书生
安装
pip install asammdf
解析
1. 读取文件,获取信号
from asammdf import MDF f = r"xxx.mdf" mdf = MDF(f) if '信号名' in mdf: #注意,一个mdf里可能有重名的信号,这个时候要加上group_index和channel_index,比如mdf.get('信号名',3,2),至于后面两个索引号怎么获取,这个就需要看后面讲的的channels_db了 signal = mdf.get('信号名') data = signal.samples timestamps = signal.timestamps
如果对mdf的格式不熟悉,可以百度下资料,本文主要介绍asammdf的使用。
2. 获取所有信号名及其索引
chn_db = mdf.channels_db
3. mdf转成dataframe
#代码接上 df = mdf.to_dataframe()
这个方法有个坑,canape采集数据允许不同信号不同频率,也就是说数据对不齐,这种情况下调这个方法会导致爆内存,所以使用时应该注意一下,如果转换时间过长,或者报错,多数是信号采集频率不同,如果事先就已经知道频率不同,只能用get方法获取单个信号,然后分析,或者用后面讲到的方法先进行数据对齐(重采样),然后在进行转换。
4. 获取channelgroup和chnannel
mdf文件一般是用channel和channel group组织的,一个文件可能包含多个chnannel group,一个channel group也可以包含多个channel,channel和signal一一对应,channel保存了一些描述信息,数据和时间戳保存在signal里,asammdf里提供了几个方法来获取信息,下面是一段从源码里摘取的一段
for group_index, (virtual_group_index, virtual_group) in enumerate( mdf_ins.virtual_groups.items() ): if virtual_group.cycles_nr == 0 and empty_channels == "skip": continue channels = [ (None, gp_index, ch_index) for gp_index, channel_indexes in mdf_ins.included_channels( virtual_group_index )[virtual_group_index].items() for ch_index in channel_indexes if ch_index != mdf_ins.masters_db.get(gp_index, None) ]
从中可以看出,我们可以通过mdf.virtual_groups获取到全部的channel group,virtual_groups是一个字典,所以可以通过items方法遍历,然后通过mdf.included_channels(virtual_group_index)方法获取到当前channel group下包含的channels,这里我们注意到,他获取的channels是一个元组列表,第一个都是None,第二个和第三个分别是channel group的index和channel的index,再看mdf.get()方法的参数其实有很多,前三个是signal name,group index,channel index,其实就是列表元组的三个,因为channel group、channel是一个二维结构,所以,其实可以不用信号名来获取信号,也可以通过group index,和channel index,然后第一个参数传None,同时,不同通道组中的通道名字可能相同,所以这时,如果仅仅传信号名,代码会报错,因为它不知道你取得哪个组里的信号,所以这时候就需要传group index和channel index,但是如果一开始就用索引就不会,因为这两个索引会唯一确定一个信号。
5. 获取一部分信号数据
基于上段的通道列表,我们就可以用mdf.select方法筛选出我们需要的信号,信号和通道是一一对应关系,我们可以把这两个东西当成一个东西理解,只不过两种数据结构存的数据不一样,所以这里大家不要混淆或者蒙圈。上面获取的其实是全部的信号,其实也可以是部分。
old_signals = [ signal for signal in mdf_ins.select( channels, raw=True, copy_master=False, validate=False ) ]
mdf.select返回的是信号列表,而不是包含这些信号的mdf实列,mdf.filter()返回的是mdf实列,这里需要注意下,当然select要保证传入的信号列表都在mdf里,否则会报错,另外不要用for循环+get('信号名')的方式获取多个信号数据,经过测试,这个函数性能很差,获取单个信号的数据还可以,信号一多最好用select,速度非常快。
还有那个raw参数,虽然它这里用的True,但是,一般我们要设置为False,特别是自己处理数据的时候,因为设置成True会导致读出的数据错误,它之所以有这个参数,我怀疑是为了兼容一些数据格式,而不是为了数据准确性。
6. 数据合并
MDF类里有个静态方法concatenate,用于多个文件的合并。
from asammdf import MDF mdf1 = MDF(r"f1.mf4") mdf2 = MDF(r"f2.mf4") mdf = MDF.concatenate([mdf1,mdf2])
concatenate是一个静态方法,可以用类直接调用,这个方法用于两个group和channel都一样的文件进行连接,即纵向连接。不支持包含不同信号的数据进行横向拼接 。
7. 数据过滤
mdf.filter(channels),这个上文已经提到,它返回的是包含指定信号的mdf实例,而不是信号列表。
8. 数据导出
asammdf支持把mdf转成其它格式,比如csv,hdf5,mat,parquet。
mdf.export(format,filename)
9. 数据绘图
asammdf支持用信号数据绘图,下面是官方示例
# map signals xs = np.linspace(-1, 1, 50) ys = np.linspace(-1, 1, 50) X, Y = np.meshgrid(xs, ys) vals = np.linspace(0, 180.0 / np.pi, 100) phi = np.ones((len(vals), 50, 50), dtype=np.float64) for i, val in enumerate(vals): phi[i] *= val R = 1 - np.sqrt(X**2 + Y**2) samples = np.cos(2 * np.pi * X + phi) * R timestamps = np.arange(0, 2, 0.02) s_map = Signal( samples=samples, timestamps=timestamps, name="Variable Map Signal", unit="dB" ) s_map.plot()
我们看到plot方法是挂在Signal上的,所以,我们用mdf.get('signame')获取到一个信号,也可以直接绘图,值得注意的是,信号是支持直接计算的,再看一个官方的例子
import numpy as np from asammdf import Signal # create 3 Signal objects with different time stamps # unit8 with 100ms time raster timestamps = np.array([0.1 * t for t in range(5)], dtype=np.float32) s_uint8 = Signal( samples=np.array([t for t in range(5)], dtype=np.uint8), timestamps=timestamps, name="Uint8_Signal", unit="u1", ) # int32 with 50ms time raster timestamps = np.array([0.05 * t for t in range(10)], dtype=np.float32) s_int32 = Signal( samples=np.array(list(range(-500, 500, 100)), dtype=np.int32), timestamps=timestamps, name="Int32_Signal", unit="i4", ) # float64 with 300ms time raster timestamps = np.array([0.3 * t for t in range(3)], dtype=np.float32) s_float64 = Signal( samples=np.array(list(range(2000, -1000, -1000)), dtype=np.int32), timestamps=timestamps, name="Float64_Signal", unit="f8", ) prod = s_float64 * s_uint8 prod.name = "Uint8_Signal * Float64_Signal" prod.unit = "*" prod.plot() pow2 = s_uint8**2 pow2.name = "Uint8_Signal ^ 2" pow2.unit = "u1^2" pow2.plot() allsum = s_uint8 + s_int32 + s_float64 allsum.name = "Uint8_Signal + Int32_Signal + Float64_Signal" allsum.unit = "+" allsum.plot() # inplace operations pow2 *= -1 pow2.name = "- Uint8_Signal ^ 2" pow2.plot() # cut signal s_int32.plot() cut_signal = s_int32.cut(start=0.2, stop=0.35) cut_signal.plot()
我们可以看到,信号是可以相乘,相加,平方操作的,操作返回应该还是信号类型,因为后面我们看到它调用了plot方法,这个方法是挂在Signal上面的。
10. 数据重采样
canape支持不同信号的采集频率不同,如果不同信号数据量不同,也就是说数据不对齐,这样在转成dataframe的时候就会爆内存,再比如频率如果过高的话,数据量必然特别大,这样的话也不利于分析,这时就需要我们对数据进行重新采样。asammdf有个resample方法。
下面是个例子:
from asammdf import MDF from utils import get_lines_from_txt f = r"C:\Users\c-master1\Desktop\下载\AnalysisTools\YIQI_2022-08-06_14-37-50.mf4" f1 = r"C:\Users\c-master1\Downloads\2022-11-21_16-59-29.mf4" f2 = r"C:\Users\c-master1\Downloads\Recorder_2022-11-17_14-55-00.MDF" clist_f = r"C:\Users\c-master1\Downloads\数据分析用信号列表(2)(2).txt" channel_list = ['ABV_Ki_Out','ACsurge_Flag_b','csABPV_posVlv_perc'] channel_list = get_lines_from_txt(clist_f) mdf = MDF(f2) mdf = mdf.filter(channel_list) signals = mdf.select(channel_list) max = 0 name = "" for sig in signals: if len(sig.samples) > max: max = len(sig.samples) name = sig.name mdf1 = mdf.resample(raster=name) df1 = mdf1.to_dataframe() df1
resample方法支持按某个信号进行重采样,即所有其它信号按这个信号对齐,也支持手动写个频率,比如0.1就是0.1秒一个数据,还支持传一个数组,采集指定时间点的数据,进行重采样后数据就对齐了,这时就可以转成dataframe了,对于升采样,有个需要注意的问题,一般mf4保存的是数字,当然也可以是字符串,对于数值型数据,主要分两种,也就是整数和小数,也叫整型和浮点型,在asammdf中,对于整数和小数的默认重采样策略是不同的,整数的话默认是重复之前的值,而对于浮点型也就是小数的话,默认是线性插值,所以,有的时候你会发现重采样后数据好像被改了,其实就是线性插值导致的,如果想改变默认行为,可以用
mdf.configure(float_interpolation=0,integer_interpolation=0)
来改变,其中mdf是MDF的实例,针对浮点型,asammdf支持两种重采样策略,一种是重复之前的值,一种就是线性插值,对于整型,提供了三种策略,前面两种和浮点型一样,还有一种混合策略,就是结合重复前值和线性插值的特点,具体没有试过。
还有一个坑需要注意,就是表面上我们看到数据里存的是整数,但是它的类型是Double的,这种asammdf也会把它当浮点型处理,因为它只按数据类型处理,不管你存的具体数据到底是整数还是小数。
mdf/mf4文件创建
1. mdf支持从零创建mdf/mf4文件
下面是官方示例
# -*- coding: utf-8 -*- """ *asammdf* MDF usage example """ import numpy as np from asammdf import MDF, Signal # create 3 Signal objects timestamps = np.array([0.1, 0.2, 0.3, 0.4, 0.5], dtype=np.float32) # unit8 s_uint8 = Signal( samples=np.array([0, 1, 2, 3, 4], dtype=np.uint8), timestamps=timestamps, name="Uint8_Signal", unit="u1", ) # int32 s_int32 = Signal( samples=np.array([-20, -10, 0, 10, 20], dtype=np.int32), timestamps=timestamps, name="Int32_Signal", unit="i4", ) # float64 s_float64 = Signal( samples=np.array([-20, -10, 0, 10, 20], dtype=np.float64), timestamps=timestamps, name="Float64_Signal", unit="f8", ) # create empty MDf version 4.00 file with MDF(version="4.10") as mdf4: # append the 3 signals to the new file signals = [s_uint8, s_int32, s_float64] mdf4.append(signals, comment="Created by Python") # save new file mdf4.save("my_new_file.mf4", overwrite=True) # convert new file to mdf version 3.10 mdf3 = mdf4.convert(version="3.10") print(mdf3.version) # get the float signal sig = mdf3.get("Float64_Signal") print(sig) # cut measurement from 0.3s to end of measurement mdf4_cut = mdf4.cut(start=0.3) mdf4_cut.get("Float64_Signal").plot() # cut measurement from start of measurement to 0.4s mdf4_cut = mdf4.cut(stop=0.45) mdf4_cut.get("Float64_Signal").plot() # filter some signals from the file mdf4 = mdf4.filter(["Int32_Signal", "Uint8_Signal"]) # save using zipped transpose deflate blocks mdf4.save("out.mf4", compression=2, overwrite=True)
当然,从官方的示例中还可以获得一些其它信息,比如,文件的裁剪,mdf.cut(**args)方法.
2. 也支持从dataframe创建
import pandas as pd import numpy as np df = pd.DataFrame({ 'a': np.arrange(10), 'b': np.arrange(10), 'c': np.arrange(10), }) mdf_f = MDF() mdf_f.append(df) mdf_f.save("xx.mf4",overwrite=True)
最后
asammdf其实还有一个gui工具,用pip install asammdf[gui]命令即可安装,图形界面实现了一些canape的功能,感觉还是很强大的,感兴趣的可以用下试试。
以上为个人经验,希望能给大家一个参考,也希望大家多多支持脚本之家。