如何将多组列表、矩阵存入单个CSV并添加专属标识？

Hey there! Let's work through your two CSV storage questions with practical, easy-to-follow solutions.

The simplest way to do this is using pandas—it handles column headers and list-to-CSV conversion seamlessly. Here's how:

Step-by-Step Example #

Suppose you have 3 lists representing different data points:

names = ["Alice", "Bob", "Charlie"]
ages = [25, 30, 35]
scores = [85, 90, 78]

1. Import pandas and create a DataFrame where each list becomes a column, with your desired headers:

import pandas as pd

# Map each list to a column name
data_dict = {
    "Name": names,
    "Age": ages,
    "Score": scores
}

df = pd.DataFrame(data_dict)

2. Save the DataFrame to a CSV file:

df.to_csv("user_data.csv", index=False, encoding="utf-8")

• index=False removes the auto-generated row numbers from the output.
• The resulting CSV will have your headers as the first row, with each list's data in the corresponding column.

If you don't want to use pandas (raw csv module) #

You can use Python's built-in csv module, though it's a bit more manual:

import csv

with open("user_data.csv", "w", newline="", encoding="utf-8") as f:
    writer = csv.writer(f)
    # Write the header row first
    writer.writerow(["Name", "Age", "Score"])
    # Zip the lists together to write each data row
    for name, age, score in zip(names, ages, scores):
        writer.writerow([name, age, score])

I get why your previous approach caused mixed data—when you just add matrices to a DataFrame directly, there's no way to distinguish which rows belong to which matrix. The fix is simple: add a unique identifier column to every row of each matrix before combining everything into one dataset.

Step-by-Step Solution #

Let's assume your matrices are 2D lists (e.g., each is m rows × n columns). Here's how to process all 30:

1. First, let's set up the example (replace this with your actual list of matrices):

import pandas as pd

# Simulate 30 matrices (3 rows × 4 columns each)
# Replace this with your own list of matrices (e.g., matrices = [temp0, temp1, ..., temp29])
matrices = []
for i in range(30):
    # Create a sample matrix (replace with your actual data)
    matrix = [
        [i*10 + 1, i*10 + 2, i*10 + 3, i*10 +4],
        [i*10 +5, i*10 +6, i*10 +7, i*10 +8],
        [i*10 +9, i*10 +10, i*10 +11, i*10 +12]
    ]
    matrices.append(matrix)

2. Process each matrix to add the identifier, then collect all rows:

all_processed_rows = []

for matrix_idx, matrix in enumerate(matrices):
    # Choose your identifier format: either a number (matrix_idx) or a string like "Matrix-{matrix_idx}"
    identifier = f"Matrix-{matrix_idx}"
    # For every row in the matrix, prepend the identifier
    for row in matrix:
        processed_row = [identifier] + row
        all_processed_rows.append(processed_row)

3. Convert to a DataFrame with clear column names and save:

# Create column names: first is the identifier, then columns for the matrix data
num_matrix_cols = len(matrices[0][0])  # Get the number of columns in your matrices
column_names = ["Matrix ID"] + [f"Col{col_num}" for col_num in range(num_matrix_cols)]

df = pd.DataFrame(all_processed_rows, columns=column_names)
# Save to CSV
df.to_csv("all_matrices.csv", index=False, encoding="utf-8")

Why this works #

Every row in the final CSV will have a Matrix ID column that tells you exactly which original matrix it came from. No more mixed data—you can easily filter or analyze rows by their matrix identifier later.

If you were using pd.concat() before without adding this identifier, that's why your data was getting mixed up. This approach ensures full traceability for each row.

内容的提问来源于stack exchange，提问作者catris25