File size: 1,742 Bytes
bf57148 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 | # CSV Sum
**Task:** "Write Python code that reads sales.csv and sums the 'amount' column."
Verbatim model output from a benchmark run, Claude Haiku 4.5, no-skill arm vs ponytail arm, temperature 1, source `benchmarks/output.json`. Reproduce: `npx promptfoo@latest eval -c benchmarks/promptfooconfig.yaml`.
## Without Ponytail, 20 lines of code
# Python code to read sales.csv and sum the 'amount' column
```python
import pandas as pd
# Read the CSV file
df = pd.read_csv('sales.csv')
# Sum the 'amount' column
total_amount = df['amount'].sum()
# Print the result
print(f"Total amount: ${total_amount:,.2f}")
```
## Alternative methods:
**Using CSV module (without pandas):**
```python
import csv
total_amount = 0
with open('sales.csv', 'r') as file:
reader = csv.DictReader(file)
for row in reader:
total_amount += float(row['amount'])
print(f"Total amount: ${total_amount:,.2f}")
```
**Using pandas with error handling:**
```python
import pandas as pd
try:
df = pd.read_csv('sales.csv')
total_amount = df['amount'].sum()
print(f"Total amount: ${total_amount:,.2f}")
except FileNotFoundError:
print("Error: sales.csv not found")
except KeyError:
print("Error: 'amount' column not found")
```
**The pandas method is recommended** because it's:
- More concise and readable
- Handles data types automatically
- Provides built-in error handling
- Efficient for large files
## With Ponytail, 3 lines of code
```python
import csv
total = sum(float(row['amount']) for row in csv.DictReader(open('sales.csv')))
print(total)
```
Skipped: pandas, error handling, file closing, add when the CSV is large, malformed, or you need more analysis.
**20 → 3 lines of code**, same model, same prompt.
|