Computer >> Máy Tính >  >> Lập trình >> Python

Thêm một cột mới vào DataFrame hiện có trong Pandas bằng Python

Trong hướng dẫn này, chúng ta sẽ tìm hiểu cách thêm một cột mới vào DataFrame hiện có ở gấu trúc. Chúng tôi có thể có các phương pháp khác nhau để thêm một cột mới. Hãy cùng tất cả chúng.

Sử dụng danh sách

Chúng tôi có thể thêm một cột mới bằng cách sử dụng danh sách. Làm theo các bước để thêm cột mới.

Thuật toán

1. Create DataFrame using a dictionary.
2. Create a list containing new column data. Make sure that the length of the list matches the length of the data which is already present in the data frame.
3. Add the list to the DataFrame like dictionary element.

Hãy xem một ví dụ.

Ví dụ

# importing pandas
import pandas as pd

# creating a DataFrame
data = {
   'Name': ['Hafeez', 'Aslan', 'Kareem'],
   'Age': [19, 18, 15],
   'Profession': ['Pythoneer', 'Programmer', 'Student']
}
dataframe = pd.DataFrame(data)
print('-----------Before adding a new column----------')
print(dataframe)
print('\n\n')

# creating a list for new column
places = ['Nellore', 'Mumbai', 'Andhra']

# we are using 'Places' as column name
# adding the list to the dataframe as column
dataframe['Places'] = places
print('---------------After adding a new column------------')
print(dataframe)

Đầu ra

Nếu bạn chạy chương trình trên, bạn sẽ nhận được kết quả sau.

-----------Before adding a new column----------
    Name     Age   Profession
0  Hafeez    19     Pythoneer
1   Aslan    18    Programmer
2  Kareem    15       Student
---------------After adding a new column------------
    Name   Age    Profession    Places
0  Hafeez   19    Pythoneer    Nellore
1   Aslan   18   Programmer     Mumbai
2 K areem   15      Student     Andhra

DataFrame.insert ()

Có một phương thức tích hợp được gọi là insert () để thêm một cột mới. Các bước cần làm theo.

Thuật toán

1. Create DataFrame using dictionary.
2. Create a list containing new column data. Make sure that the length of the list matches the length of the data which is already present in the data frame.
3. Insert the data into the DataFrame using DataFrame.insert(index, column_name, data)
method.

Ví dụ

# importing pandas
import pandas as pd

# creating a DataFrame
data = {
   'Name': ['Hafeez', 'Aslan', 'Kareem'],
   'Age': [19, 18, 15],
   'Profession': ['Pythoneer', 'Programmer', 'Student']
}
dataframe = pd.DataFrame(data)
print('-----------Before adding a new column----------')
print(dataframe)
print('\n\n')

# creating a list for new column
places = ['Nellore', 'Mumbai', 'Andhra']

# we are using 'Places' as column name
# adding the list to the dataframe as column using insert(index, column_name, data)
dataframe.insert(2, 'Places', places)
print('---------------After adding a new column------------')
print(dataframe)

Đầu ra

Nếu bạn chạy chương trình trên, bạn sẽ nhận được kết quả sau.

-----------Before adding a new column----------
      Name  Age  Profession
0   Hafeez   19   Pythoneer
1    Aslan   18  Programmer
2   Kareem   15     Student

---------------After adding a new column------------
     Name  Age   Places  Profession
0  Hafeez   19  Nellore   Pythoneer
1   Aslan   18   Mumbai  Programmer
2  Kareem   15   Andhra     Student

DataFrame.assign ()

Phương thức này nhận một đối số, tức là .., một danh sách dữ liệu và thêm nó vào khung dữ liệu dưới dạng một cột ở cuối.

Thuật toán

1. Create DataFrame using a dictionary.
2. Create a list containing new column data. Make sure that the length of the list matches the length of the data which is already present in the data frame.
3. Insert the data into the DataFrame using DataFrame.assign(column_name = data)
method. It returns a new data frame. So, we have to store it.
4. Print the new data frame.

Hãy xem một ví dụ.

Ví dụ

# importing pandas
import pandas as pd
# creating a DataFrame
data = {
   'Name': ['Hafeez', 'Aslan', 'Kareem'],
   'Age': [19, 18, 15],
   'Profession': ['Pythoneer', 'Programmer', 'Student']
}
dataframe = pd.DataFrame(data)
print('-----------Before adding a new column----------')
print(dataframe)
print('\n\n')
# creating a list for new column
places = ['Nellore', 'Mumbai', 'Andhra']
# we are using 'Places' as column name
# adding the list to the dataframe as column using assign(column_name = data)
new_dataframe = dataframe.assign(Places = places)
print('---------------After adding a new column------------')
print(new_dataframe)

Đầu ra

Nếu bạn chạy chương trình trên, bạn sẽ nhận được kết quả sau.

-----------Before adding a new column----------
   Name     Age     Profession
0  Hafeez   19       Pythoneer
1   Aslan   18      Programmer
2  Kareem   15         Student
---------------After adding a new column------------
     Name    Age    Profession    Places
0  Hafeez    19      Pythoneer   Nellore
1   Aslan    18     Programmer    Mumbai
2  Kareem    15        Student    Andhra

Kết luận

Nếu bạn có bất kỳ nghi ngờ nào về hướng dẫn, hãy đề cập đến chúng trong phần bình luận.