擅长:python、mysql、java
<h2>使用<code>strip</code>、groupby和Pandas删除空格:</h2>
<p>输入文件(可选空格是有意的):</p>
<pre><code>author,books
Vincent, 1
Vincent , 1
Vincent, 1
Vincent, 1
Thomas , 1
Thomas, 1
Thomas, 1
Jimmy, 1
Jimmy , 1
import csv
import pandas as pd
fin = open('author.csv', 'r')
reader = csv.DictReader(fin, delimiter=',')
# strip remove spaces
authors=[( (d['author']).strip(), int((d['books']).strip())) for d in reader]
df = pd.DataFrame(authors)
df.columns = ['author', 'books']
df2 = (df.groupby('author').sum())
print (df2)
books
author
Jimmy 2
Thomas 3
Vincent 4
# For total of books:
print (df2.books.sum())
9
</code></pre>