如何在Python中移除DataFrame的初始零值行（保留中间零值）

Got it, let's tackle this problem head-on. You need to strip off those initial zero-sales rows from each of your 4 product DataFrames, but keep any zeros that pop up in the middle of the weekly data. Here's a straightforward, efficient way to do this with pandas:

Step 1: Process a Single DataFrame #

First, let's walk through handling one product's data using your Product 1 example as a test case:

import pandas as pd

# Sample Product 1 DataFrame
df_product1 = pd.DataFrame({
    'week': ['周1', '周2', '周3', '周4', '周5', '周6', '周7'],
    'sales': [0, 0, 100, 120, 55, 0, 60]
})

# Find the first row where sales isn't zero
first_non_zero_idx = df_product1[df_product1['sales'] != 0].index[0]

# Slice the DataFrame to keep everything from that row onward
cleaned_df = df_product1.loc[first_non_zero_idx:]

print(cleaned_df)

Running this will output exactly what you want: it drops the first two zero rows, but keeps the zero in week 6.

Step 2: Batch Process All 4 Product DataFrames #

If you have all your product DataFrames stored in a dictionary (a clean way to organize multiple similar datasets), you can loop through them to clean each one efficiently:

# Store all your product DataFrames in a dictionary
product_dfs = {
    'product1': df_product1,
    'product2': pd.DataFrame({
        'week': ['周1', '周2', '周3', '周4', '周5', '周6', '周7'],
        'sales': [0, 0, 0, 120, 0, 30, 60]
    }),
    # Add product3 and product4 DataFrames here
}

# Clean each DataFrame and store the results
cleaned_product_dfs = {}
for product_name, df in product_dfs.items():
    # Get all rows with non-zero sales
    first_non_zero_rows = df[df['sales'] != 0]
    # Add a safety check in case a DataFrame has all zeros (optional, based on your data)
    if not first_non_zero_rows.empty:
        first_non_zero_idx = first_non_zero_rows.index[0]
        cleaned_product_dfs[product_name] = df.loc[first_non_zero_idx:]
    else:
        # Handle all-zero DataFrames if needed (e.g., keep them or skip)
        cleaned_product_dfs[product_name] = df

# Check the cleaned Product 2 DataFrame
print(cleaned_product_dfs['product2'])

This will clean Product 2 by removing the first three zero rows, while preserving the zero in week 5.

Key Notes #

• This method only removes leading consecutive zeros—any zeros that come after the first non-zero sales value stay intact, which matches your requirement perfectly.
• The safety check for all-zero DataFrames is optional, but it prevents errors if one of your products has no sales records at all.

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