如何将用户数据添加到 Excel 文件？

Question

我有一个带有现成工作表的现成文件，我只需要在其中添加 2 列的数据，请帮助

| Teachers       | Subjects       |
| -------------- | -------------- |
| TeachersName   | SubjectsName   |
| TeachersName   | SubjectsName   |
| TeachersName   | SubjectsName   |
| TeachersName   | SubjectsName   |

def teach_add():
df = pd.read_excel(open('bd1.xlsx','rb'),sheet_name='teachers')
print(df[['Teachers','Subjects']])
mas = []
num = 2
df['Teachers'] = mas
ind  = int(input("Enter index: "))
for i in range(0, num):
    num2 = num2.split(' ')
    mas.append(num2)
print(mas)
df.loc[ind]=[mas]
df.to_excel('bd1.xlsx', sheet_name='teachers')

Answer 1

我不知道您的档案中有什么 - 但如果您已经有行，那么对于单个教师和科目可能会简单得多

def teach_add():
    # read data from file
    df = pd.read_excel('bd1.xlsx', sheet_name='teachers')

    #print(df[['Teachers','Subjects']])
    
    # ask for data
    index  = int(input("Enter index: "))
    teacher = input("Enter teacher: ")
    subject = input("Enter subject: ")

    # add data to dataframe
    df['Teachers'][index] = teacher
    df['Subjects'][index] = subject

    # save data to file
    df.to_excel('bd1.xlsx', sheet_name='teachers')

这里是代码中包含一些数据的最小工作代码 - 所以每个人都可以简单地复制它并运行它来看看它是如何工作的

import pandas as pd

df = pd.DataFrame({
            'Class': ['A', 'B', 'C'], 
            'Teachers':['', '', ''], 
            'Subjects':['', '', ''], 
    })

#print(df[['Teachers','Subjects']])
print(df)

index   = 1
teacher = 'Albert Einstein'
subject = 'Physics'

df['Teachers'][index] = teacher
df['Subjects'][index] = subject

index   = 2
teacher = 'Alfred Nobel'
subject = 'Chemistry'

df['Teachers'][index] = teacher
df['Subjects'][index] = subject

#print(df[['Teachers','Subjects']])
print(df)

之前

  Class Teachers Subjects
0     A                  
1     B                  
2     C  

之后

  Class         Teachers   Subjects
0     A                            
1     B  Albert Einstein    Physics
2     C     Alfred Nobel  Chemistry

---

如果你想要求更多的老师，那么你应该使用for-loop。你可以问你要加多少老师

def teach_add():
    # read data from file
    df = pd.read_excel('bd1.xlsx', sheet_name='teachers')

    #print(df[['Teachers','Subjects']])
    
    # how many rows you have in dataframe
    teachers_number = int(input("How many teachers: "))

    # --- loop ---

    for _ in range(teachers_number):

        # ask for data
        index   = int(input("Enter index: "))
        teacher = input("Enter teacher: ")
        subject = input("Enter subject: ")

        # add data to dataframe
        df['Teachers'][index] = teacher
        df['Subjects'][index] = subject


    # --- after loop ---

    # save data to file
    df.to_excel('bd1.xlsx', sheet_name='teachers')

或者你可以得到你有多少空行并填充所有行而不要求索引。

def teach_add():
    # read data from file
    df = pd.read_excel('bd1.xlsx', sheet_name='teachers')

    #print(df[['Teachers','Subjects']])
    
    # how many rows you have in dataframe
    rows_number = len(df)

    # --- loop ---

    for index in range(rows_number):

        # ask for data
        teacher = input("Enter teacher: ")
        subject = input("Enter subject: ")

        # add data to dataframe
        df['Teachers'][index] = teacher
        df['Subjects'][index] = subject

    # --- after loop ---

    # save data to file
    df.to_excel('bd1.xlsx', sheet_name='teachers')

如果您在 DataFarme 中没有行并且您必须追加行，它可能还需要不同的代码。

---

编辑：

pandas 警告我它适用于数据副本，我必须使用

df.loc[index, 'Teachers'] = teacher
df.loc[index, 'Subjects'] = subject

而不是

df['Teachers'][index] = teacher
df['Subjects'][index] = subject

我还必须使用 read_excel(..., index=False) 和 to_excel(..., index=False) 来跳过添加带索引的列。