Замена значения столбца CSV-файла в Python

Опубликовано: 27 Марта, 2022

Давайте посмотрим, как мы можем заменить значение столбца файла CSV в Python. CSV-файл - это не что иное, как файл с разделителями-запятыми.

Метод 1: использование нативного языка Python

Используя метод replace () , мы можем легко заменить текст другим текстом. В приведенном ниже коде у нас будет входной CSV-файл как «csvfile.csv», который будет открыт в режиме «чтения». Метод join () берет все строки файла CSV в итерируемом объекте и объединяет их в одну строку. Затем мы можем использовать метод replace () для всей строки и выполнять одну / несколько замен. Во всей строке ищется данный текст и заменяется указанным текстом.

Пример:

Входной файл будет:

Python3

# reading the CSV file
text = open("csvfile.csv", "r")
  
#join() method combines all contents of 
# csvfile.csv and formed as a string
text = "".join([i for i in text]) 
  
# search and replace the contents
text = text.replace("EmployeeName", "EmpName"
text = text.replace("EmployeeNumber", "EmpNumber"
text = text.replace("EmployeeDepartment", "EmpDepartment"
text = text.replace("lined", "linked"
  
# output.csv is the output file opened in write mode
x = open("output.csv","w")
  
# all the replaced text is written in the output.csv file
x.writelines(text)
x.close()

Выход:

Метод 2: Использование Pandas DataFrame

Мы можем прочитать CSV-файл как DataFrame, а затем применить метод replace () .

Python3

# importing the module
import pandas as pd 
    
# making data frame from the csv file 
dataframe = pd.read_csv("csvfile1.csv"
    
# using the replace() method
dataframe.replace(to_replace ="Fashion"
                 value = "Fashion industry"
                  inplace = True)
dataframe.replace(to_replace ="Food"
                 value = "Food Industry"
                  inplace = True)
dataframe.replace(to_replace ="IT"
                 value = "IT Industry"
                  inplace = True)
  
# writing  the dataframe to another csv file
dataframe.to_csv("outputfile.csv"
                 index = False)

Выход:

Внимание компьютерщик! Укрепите свои основы с помощью базового курса программирования Python и изучите основы.

Для начала подготовьтесь к собеседованию. Расширьте свои концепции структур данных с помощью курса Python DS. А чтобы начать свое путешествие по машинному обучению, присоединяйтесь к курсу Машинное обучение - базовый уровень.