I am trying to write a Python script that reads a CSV file with zip codes, fetches zip codes within a radius using an API, and then populates the results into a new column in the CSV. The API requests seem to be working correctly, and I can see the responses in the console output. However, the resulting CSV file does not have the expected values in the radius_zips column.
Here is my current script:
import pandas as pdimport requests# Define the file pathsinput_file_path = 'test_zips.csv' # Located on the Desktopoutput_file_path = 'test_zips_with_radius_zips_output.csv' # Will be saved on the Desktop# Define the API detailsurl = "https://zip-code-distance-radius.p.rapidapi.com/api/zipCodesWithinRadius"headers = {"x-rapidapi-key": "your_api_key","x-rapidapi-host": "zip-code-distance-radius.p.rapidapi.com"}# Read the CSV filedf = pd.read_csv(input_file_path)# Function to get zip codes within a radius for a given zip codedef get_radius_zips(zip_code, radius="10"): querystring = {"zipCode": zip_code, "radius": radius} try: response = requests.get(url, headers=headers, params=querystring) response.raise_for_status() data = response.json() print(f"Response for {zip_code}: {data}") # Print the full response if 'zip_codes' in data: zip_codes = [item['zipCode'] for item in data] print(f"Zip codes within radius for {zip_code}: {zip_codes}") return ', '.join(zip_codes) else: print(f"No zip codes found for {zip_code}") except requests.exceptions.RequestException as e: print(f"Error fetching data for zip code {zip_code}: {e}") except ValueError as e: print(f"Error parsing JSON response for zip code {zip_code}: {e}") return ''# Apply the function to the total_zips column and create the radius_zips columndef process_total_zips(total_zips): zip_codes = total_zips.split(', ') radius_zip_codes = [get_radius_zips(zip.strip()) for zip in zip_codes] radius_zip_codes = [z for z in radius_zip_codes if z] # Filter out empty strings return ', '.join(radius_zip_codes) if radius_zip_codes else ''df['radius_zips'] = df['total_zips'].apply(process_total_zips)# Write the modified DataFrame to a new CSV filedf.to_csv(output_file_path, index=False)print("The new CSV file 'test_zips_with_radius_zips_output.csv' has been created.")The console output indicates that the API responses are correct:
Response for 01001: [{'zipCode': '01001', 'distance': 0.0}, {'zipCode': '01106', 'distance': 2.9825640831681617}, ...]Zip codes within radius for 01001: ['01001', '01106', ...]However, the resulting CSV file still has an empty radius_zips column.
Here is a sample of the input CSV file (test_zips.csv):
lat,lng,city,state_id,state_name,population,density,shortcode,total_zips42.06262,-72.62521,Agawam,MA,Massachusetts,16045,548.6,ma_agawam,0100142.37633,-72.46462,Amherst,MA,Massachusetts,22992,166.7,ma_amherst,01002,01003...Here is a sample of the incorrect output CSV file (test_zips_with_radius_zips_output.csv):
lat,lng,city,state_id,state_name,population,density,shortcode,total_zips,radius_zips42.06262,-72.62521,Agawam,MA,Massachusetts,16045,548.6,ma_agawam,01001,42.37633,-72.46462,Amherst,MA,Massachusetts,22992,166.7,ma_amherst,01002,01003,...What could be causing the radius_zips column to be empty in the output CSV, even though the API responses are correct?