python批量读取raw格式文件并进行修改
时间: 2023-10-18 10:31:32 浏览: 139
要批量读取raw格式文件并进行修改,可以使用Python中的`os`和`numpy`模块。下面是一个简单的例子,假设我们要将一个raw格式的文件中所有像素的值加上100:
```python
import os
import numpy as np
# 定义要读取的目录和文件类型
data_dir = './data'
file_type = '.raw'
# 遍历目录下的所有文件
for filename in os.listdir(data_dir):
if filename.endswith(file_type):
# 读入raw文件
with open(os.path.join(data_dir, filename), 'rb') as f:
data = np.fromfile(f, dtype=np.uint8)
# 修改数据
data += 100
# 保存修改后的数据
with open(os.path.join(data_dir, filename), 'wb') as f:
data.tofile(f)
```
这个例子中,我们首先定义了要读取的目录和文件类型,然后使用`os.listdir`遍历目录下的所有文件,对于每个以`.raw`为后缀的文件,我们使用`numpy.fromfile`读入数据,将数据加上100,然后使用`numpy.tofile`保存修改后的数据。注意,在使用`numpy.fromfile`和`numpy.tofile`时,需要指定数据类型和字节顺序(在本例中,我们使用了默认值)。
阅读全文