我有一个Pandas DataFrame,它是用tabula py从PDF中提取的。
PDF是这样的:
+--------------+--------+-------+
| name | letter | value |
+--------------+--------+-------+
| A short name | a | 1 |
+-------------------------------+
| Another | b | 2 |
+-------------------------------+
| A very large | c | 3 |
| name | | |
+-------------------------------+
| other one | d | 4 |
+-------------------------------+
| My name is | e | 5 |
| big | | |
+--------------+--------+-------+
正如您所看到的,A very large name
有一个换行符,并且由于原始pdf没有边框,所以在DataFrame中创建了一个带有['name', NaN, NaN]
的行和另一个带['A very large', 'c', 3]
的行,而我只想要一个带有内容的行:['A very large name', 'c', 3]
。
My name is big
也是如此
由于这发生在几行中,我试图实现的是当该行中的其他单元格为NaN
时,将name
单元格的内容与前一个单元格连接起来。然后删除NaN行。
但任何其他获得同样结果的策略都是受欢迎的。
import pandas as pd
import numpy as np
data = {
"name": ["A short name", "Another", "A very large", "name", "other one", "My name is", "big"],
"letter": ["a", "b", "c", np.NaN, "d", "e", np.NaN],
"value": [1, 2, 3, np.NaN, 4, 5, np.NaN],
}
df = pd.DataFrame(data)
data_expected = {
"name": ["A short name", "Another", "A very large name", "other one", "My name is big"],
"letter": ["a", "b", "c", "d", "e"],
"value": [1, 2, 3, 4, 5],
}
df_expected = pd.DataFrame(data_expected)
我正在尝试这样的代码,但不起作用
# Not works and not very `pandastonic`
nan_indexes = df[df.iloc[:, 1:].isna().all(axis='columns')].index
df.loc[nan_indexes - 1, "name"] = df.loc[nan_indexes - 1, "name"].str.cat(df.loc[nan_indexes, "name"], ' ')
# remove NaN rows
根据列的不同,您可以尝试使用groupby.agg
和join
或first
。通过检查列字母和值中的CCD_ 11和CCD_。
print (df.groupby(df[['letter', 'value']].notna().any(1).cumsum())
.agg({'name': ' '.join, 'letter':'first', 'value':'first'})
)
name letter value
1 A short name a 1.0
2 Another b 2.0
3 A very large name c 3.0
4 other one d 4.0
5 My name is big e 5.0