数据帧示例
索引 | fileName | 起始行结束行 | |
---|---|---|---|
0 | 293104.java | 30 | 40 |
1 | 2288951.java | 183 | <247>|
2 | 2378709.java | 98 | 117
如果使用apply
,它将把函数应用到每一行,那么编写函数以获取数据帧的一行,然后可以使用点表示法访问函数中的列。
def snippetMaker(row):
file = open(row.fileName,'r').read()
snippet = file.split('n')[row.startLine:row.endLine]
cleanSnippet = str(snippet).replace('[','').replace(']','').replace(',',' ')
return cleanSnippet
df['snippet'] = df.apply(snippetMaker, axis=1)