Python中如何将文件循环计算结果转为指定结构的DataFrame？

Got it, let's get your results DataFrame working the way you want! Here's a straightforward, actionable solution based on what you described.

First, let's break down the process into simple, reproducible steps:

• Iterate through your files and run your calculation on each one
• Keep track of which filename maps to which calculation results
• Convert that mapped data into a pandas DataFrame with filenames as row labels

Example Working Code #

Here's a complete implementation you can adapt to your actual calculation logic:

import pandas as pd
import os

# Replace this with your actual calculation function
def run_calculation(file_path):
    # Example: Let's calculate file size and line count (swap this for your logic)
    file_size = os.path.getsize(file_path)
    with open(file_path, 'r') as f:
        line_count = len(f.readlines())
    # Return results as a dict (easy to turn into DataFrame columns)
    return {'file_size': file_size, 'line_count': line_count}

# Point this to your directory of files
file_directory = '/path/to/your/target/files'
# Filter for specific file types if needed (e.g., '.csv', '.txt')
all_files = [f for f in os.listdir(file_directory) if os.path.isfile(os.path.join(file_directory, f))]

# Collect filename + calculation pairs
results_collection = []
for filename in all_files:
    full_path = os.path.join(file_directory, filename)
    calc_output = run_calculation(full_path)
    # Attach the filename to the result dict
    calc_output['filename'] = filename
    results_collection.append(calc_output)

# Convert to DataFrame and set filename as row index
results = pd.DataFrame(results_collection)
results = results.set_index('filename')

# If you only have one calculation value per file, simplify with a Series:
# results = pd.Series({f: run_calculation(os.path.join(file_directory, f)) for f in all_files}).to_frame(name='calculation_result')

Troubleshooting Your Current Code #

If your existing code isn't working, check for these common issues:

• Missing link between filenames and results: Don't store filenames and calculations in separate unconnected lists—keep them paired (like in the dict above)
• Incorrect index setup: Make sure you explicitly set the filename as the index using set_index() or by passing the index parameter when creating the DataFrame
• Mismatched result structures: If your calculation returns multiple values, ensure each entry in your collection is a consistent structure (dict or list) so pandas can map them to columns correctly

Verify the Output #

Check that your DataFrame looks right with:

print(results.head())

You should see filenames as the leftmost row identifiers, with your calculation values in the columns.

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