python:txt写入csv:abaqus二次开发rpy文件数据提取

402 阅读3分钟

本文已参与「新人创作礼」活动,一起开启掘金创作之路。

本文章适合编程经验较少的人参考学习

问题描述

使用脚本控制abaqus批量构建RVE模型(调用micro mechanics插件)并且使用EasyPBC添加边界约束得到E11与V21结果,完整的所有结果包含在rpy文件中,因此需要将结果信息提取出来,放入csv文件中。

方法一

之前的方法在下面,但是比较繁琐。经过学习后得到该方法,相比之前的方法来说不需要文件处理。

直接拿到abaqus输出的rpy文件重命名为txt。观察到E11与V21的数据具有固定的格式,因此灵活使用.split功能将E11与V21数据中的数字提取出来,再放进csv文件即可。

image.png

import numpy as np
import csv

# 使用list存放数据
list = []

# 使用with open可以不用繁琐地使用close()
# 先进行txt文件数据的读取
with open('abaqus.txt', 'r') as f:
    
    # enumerate() 函数用于将一个可遍历的数据对象(如列表、元组或字符串)组合为一个索引序列,同时列出数据下标和数据
    # row_num为当前行数,line为当前行数据
    for row_num, line in enumerate(f):
        
        # 使用两次split将前后无用的字符处理掉,注意每次split后line就会变成列表形式,所以注意脚标
        if '#: E11' in line:
            line = line.split('#: E11=')
            line = line[1].split(' Stress units')
            list.append(line[0])

        if '#: V21' in line:
            line = line.split('#: V21=')
            line = line[1].split(' ratio')
            list.append(line[0])
            
# 为了写入csv时为两列,使用array的reshape功能,将list变为两列的arr
arr = np.array(list).reshape(-1, 2)

# newline=''可避免写入csv后出现空白行
with open("test.csv", "w", newline='') as csvfile:
    writer = csv.writer(csvfile)
    writer.writerows(arr)

方法二

此方法比较繁琐,不推荐使用

txt文件预处理

rpy文件直接重命名为txt文件方便操作。 为简化代码操作,首先对txt文件进行简单的处理。 原rpy文件如下图所示,在图中最后两行才是需要的数据,由于批量执行的规律性,发现从第一次有效数据开始,每隔47行为下组数据。

在这里插入图片描述 因此,删除第一组数据之前的无效代码。观察到,数据后包含“ Stress units”与“ ratio”无效后缀,因此可用文件中“替代”操作删除该后缀。得到新txt文件如图所示。

在这里插入图片描述

python代码

代码与注释如下

import numpy as np
import csv

# 使用list存放数据
list = []

# 使用with open可以不用繁琐地使用close()
# 先进行txt文件数据的读取
with open('test out.txt', 'r') as f:
    
    # enumerate() 函数用于将一个可遍历的数据对象(如列表、元组或字符串)组合为一个索引序列,同时列出数据下标和数据
    # row_num为当前行数,line为当前行数据
    for row_num, line in enumerate(f):
        
        # 第1行与之后的i+47行包含E11数据
        if row_num % 47 == 0:
            # 可使用split函数进行数据提取,根据上图,数据包含在line[1]中
            line = line.split('#: E11=')
            # 由于该方法提取数据后,line[1]最后有字符'/n',所以使用[:-1],保留除了最后的字符的数据。添加至list中
            list.append(line[1][:-1])
            
        # 这里V12直接添加到E11后面了
        if row_num % 47 == 1:
            line = line.split('#: V12=')
            list.append(line[1][:-1])

# 为了写入csv时为两列,使用array的reshape功能,将list变为两列的arr
arr = np.array(list).reshape(-1, 2)

# newline=''可避免写入csv后出现空白行
with open("test.csv", "w", newline='') as csvfile:
    writer = csv.writer(csvfile)
    writer.writerows(arr)