间隔任意行读取csv文件的数据

import numpy as np
import matplotlib.pyplot as plt
import pandas as pd
import csv



# 输入和输出文件路径
input_csv_file = '---output_data.csv'  # 输入CSV文件路径
output_csv_file = '----20241011output_data.csv'  # 输出CSV文件路径
interval = 100   # 间隔行数，例如每隔2行读取一次数据

# 读取CSV文件并保存间隔行数据到新的CSV文件
with open(input_csv_file, 'r', newline='') as infile, open(output_csv_file, 'w', newline='') as outfile:
    reader = csv.reader(infile)
    writer = csv.writer(outfile)

    # 读取数据并间隔处理
    for i, row in enumerate(reader):
        if i % interval == 0:
            writer.writerow(row)


data = pd.read_csv(output_csv_file)
data = np.array(data)
print(data.shape)
# print(data)

# Plot the matrix as an image
plt.imshow(data, cmap='jet')  # Use 'viridis' colormap for better visualization
plt.colorbar()  # Add a color bar for reference
plt.title('Matrix Visualization')  # Add a title
plt.xlabel('X-axis')  # Add label for x-axis
plt.ylabel('Y-axis')  # Add label for y-axis
plt.show()