我有一个包含 3 列的 Excel 文件。
请举例如下
Name Produce Number
Adam oranges 6
bob Apples 5
Adam Apples 4
steve Peppers 7
bob Peppers 16
Adam oranges 5
我需要以这种方式在 python 中生成总和
Name Produce Number
Adam oranges 11
bob apples 5
steve peppers 7
etc
我是Python新手,想弄清楚如何编写每个人总计的输出?有没有一种简单的方法可以从Excel中收集这些数据?
这是它的代码:
如果您有任何疑问(或者我犯了任何错误),请务必查看评论并发回此处
import csv
file = open('names.csv', "rb") #Open CSV File in Read Mode
reader = csv.reader(file) #Create reader object which iterates over lines
class Object: #Object to store unique data
def __init__(self, name, produce, amount):
self.name = name
self.produce = produce
self.amount = amount
rownum = 0 #Row Number currently iterating over
list = [] #List to store objects
def checkList(name, produce, amount):
for object in list: #Iterate through list
if object.name == name and object.produce == produce: #Check if name and produce combination exists
object.amount += int(amount) #If it does add to amount variable and break out
return
newObject = Object(name, produce, int(amount)) #Create a new object with new name, produce, and amount
list.append(newObject) #Add to list and break out
for row in reader: #Iterate through all the rows
if rownum == 0: #Store header row seperately to not get confused
header = row
else:
name = row[0] #Store name
produce = row[1] #Store produce
amount = row[2] #Store amount
if len(list) == 0: #Default case if list = 0
newObject = Object(name, produce, int(amount))
list.append(newObject)
else: #If not...
checkList(name, produce, amount)
rownum += 1
for each in list: #Print out result
print each.name, each.produce, each.amount
file.close() #Close file
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)