es.davy.ai

Preguntas y respuestas de programación confiables

¿Tienes una pregunta?

Si tienes alguna pregunta, puedes hacerla a continuación o ingresar lo que estás buscando.

Cómo leer todos los archivos CSV que comienzan con una consonante?

import os
for file in os.listdir(“/content/drive/MyDrive/BigData2021/Lecture23/datasets”):
if file.endswith(“.csv”):
print(os.path.join(file))

cities.csv
airports.csv
data_scientist_salaries.csv

Quiero leer este archivo CSV con spark que comience con una consonante sin especificar el nombre del archivo CSV. ¿Cómo se hace eso?

Tags:  , ,

Answer

  1. Avatar for davy.ai

    Para leer un archivo CSV con Spark que comienza con una consonante sin especificar el nombre de archivo, puedes usar el módulo os y la función startswith() para buscar archivos CSV en el directorio que comiencen con una consonante. Esto se puede hacer de la siguiente manera:

    import os
    from pyspark.sql import SparkSession
    
    # Crear una SparkSession
    spark = SparkSession.builder.appName("Read_CSV").getOrCreate()
    
    # Establecer la ruta al directorio donde se encuentran los archivos CSV
    csv_path = "/content/drive/MyDrive/BigData2021/Lecture23/datasets"
    
    # Recorrer los archivos en el directorio y buscar archivos CSV
    for file in os.listdir(csv_path):
        # Verificar si el archivo es un archivo CSV y comienza con una consonante
        if file.endswith(".csv") and file[0] not in "aeiouAEIOU":
            # Leer el archivo CSV en un dataframe de Spark
            df = spark.read.options(header=True, inferSchema=True).csv(os.path.join(csv_path, file))
            # Hacer algo con el dataframe (por ejemplo, imprimir el esquema)
            df.printSchema()
    

    Este código buscará archivos CSV en el directorio datasets que comiencen con una consonante, y por cada archivo que encuentre, leerá el archivo en un dataframe de Spark y hará algo con él (en este ejemplo, imprimirá el esquema del dataframe).

Comments are closed.