personal-projects/leo-claude-mktplace
Template
0
1
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix(data-platform): reset index after filter to prevent extra column

Browse Source 
The filter tool was adding an __index_level_0__ column to results
because pandas query() preserves the original index, which gets
converted to a column when storing the DataFrame.

Added .reset_index(drop=True) after query() to drop the preserved
index and create a clean sequential index.

Fixes #203

Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>
This commit is contained in:
Leo Miranda
2026-01-26 17:25:37 -05:00
parent c5232bd7bf
commit 4ed3ed7e14
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
mcp-servers/data-platform/mcp_server/pandas_tools.py
View File

@@ -330,7 +330,7 @@ class PandasTools:
return {'error': f'DataFrame not found: {data_ref}'} return {'error': f'DataFrame not found: {data_ref}'}
try: try:
filtered = df.query(condition) filtered = df.query(condition).reset_index(drop=True)
result_name = name or f"{data_ref}_filtered" result_name = name or f"{data_ref}_filtered"
return self._check_and_store( return self._check_and_store(
filtered, filtered,