如何在Python中使用.writerow()切片

时间:2013-01-01 17:23:00

标签: python datetime csv python-2.7

我遇到writerow()功能问题。我有一个像这样的对象(orders_matrix):

[[datetime.datetime(2008, 11, 20, 16, 0), 'AA', 'Buy', '100'],     
[datetime.datetime(2008, 11, 28, 16, 0), 'AA', 'Sell', '100'],
[datetime.datetime(2009, 2, 17, 16, 0), 'AA', 'Buy', '100']]

并希望以CSV格式编写,如下所示:

2008,11,20,AA,Buy,100,
2008,11,28,AA,Sell,100,
2009,2,17,AA,Buy,100,

我尝试在for循环中使用writerow()函数:

with open('orders.csv', 'w') as orders_file:
  writer = csv.writer(orders_file)
  for row in orders_matrix:
    writer.writerow([row[:4], row[5:7], row[8:10], row[11:]])

但无法转换datetime对象并添加其他元素。我想知道是否可以这样做以及如何做到。

2 个答案:

答案 0 :(得分:5)

您可以使用datetime对象的年,月,日属性来获取相关值。因此,最后一行可能会改为:

writer.writerow([row[0].year, row[0].month, row[0].day, row[1], row[2], row[3]])

答案 1 :(得分:2)

您可以使用datetime.timetupledatetime对象转换为struct_time对象,然后在将其转换为列表后切出所需信息

with open("orders.csv", "wb") as fin:
    writer = csv.writer(fin)
    for row in orders_matrix:
        writer.writerow(list(row[0].timetuple())[:3] + row[1:])