TDS Assignments | IIT Madras - Tools in Data Science Graded Assignments Answers

Python Code Generator: Pandas Analysis

Select Dataset

Context: E-commerce sales transactions

Columns: order_id, customer_id, product_category, quantity, unit_price, order_date

Analysis: total revenue by product category

Generated Python Code

import pandas as pd

# Load the data
df = pd.read_csv('sales_data.csv')

# Calculate derived column (revenue)
df['revenue'] = df['quantity'] * df['unit_price']

# Group by and aggregate
result = df.groupby('product_category')['revenue'].sum()

# Display result
print(result)