• 闪存
• 博客
• 发言 小组
• 投递 新闻
• 提问 博问
• 添加 收藏
• 文库

python 处理csv文件的两列数据，并将对比结果写入另一列

0

[待解决问题]

我现在在做数据处理，这是一个csv文件，想用实现如图的效果：

第2列的1对应第4列1101，那么在第五列标记为1

第2列的2对应第4列1102，那么在第五列标记为2

第2列的3对应第4列1101，那么在第五列标记为1

其他情况标记错的，那么在第五列标记为0

请大神帮忙使用pathon如何实现这样的效果。

ITCSJ | 菜鸟二级 | 园豆：252
提问于：2018-10-19 10:31

显示帮助

使用"Ctrl+Enter"可进行快捷提交，评论支持部分 Markdown 语法：[link](http://example.com) _italic_ **bold** `code`。

< >

分享

分享您的问题

# -*- coding: utf-8 -*-

import numpy as np

A = np.loadtxt('d:\\test.cvs', delimiter = ',', usecols = (3, 4))
rowcount = np.shape(A)[0]
B = np.zeros(rowcount).reshape(4, 1)

for i in range(rowcount):
    if A[i, 0] == 1101 and (A[i, 1] == 1 or A[i, 1] == 3):
        B[i, 0] = 1
    elif A[i, 0] == 1102 and A[i, 1] == 2:
        B[i, 0] = 2

C = np.hstack((A, B))

np.savetxt('D:\\result.cvs', C, delimiter = ',')