Как вернуть элемент альтернативного столбца из команды intersect?

Question

В настоящее время я использую следующий код, чтобы получить столбец даты пересечения двух наборов финансовых данных. Массивы включают в себя дату, O, H, L, кл

#find intersection of date strings 
def intersect(seq1, seq2): 
    res = []      # start empty 
    for x in seq1:    # scan seq1 
     if x in seq2:   # common item? 
      res.append(x) 


    return res 


x = intersect(seta[:,0], setb[:,0]) # mixed types 
print x 

Проблема заключается в том, что возвращает только столбец, для которого он нашел пересечение обоих, а именно столбец даты. Я бы хотел, чтобы он каким-то образом возвращал другой массив столбцов , включающий как значения cls для каждого набора ... т.е. , если дата является общей для обоих возвратов массива 2X1 из двух соответствующих столбцов cls . Есть идеи? Благодарю.

Answer 1

Хорошо, вот полное решение.

Получить python library to download stocks quotes

Получить некоторые цитаты

start_date, end_date = '20090309', '20090720' 
ibm_data = get_historical_prices('IBM', start_date, end_date) 
msft_data = get_historical_prices('MSFT', start_date, end_date) 

Преобразовать строки в дата-шпонкой словарей словарей

def quote_series(series): 
    columns = ['open', 'high', 'low', 'close', 'volume'] 
    return dict((item[0], dict(zip(columns, item[1:]))) for item in series[1:]) 

ibm = quote_series(ibm_data) 
msft = quote_series(msft_data) 

ли пересечение дат вещь

ibm_dates = set(ibm.keys()) 
msft_dates = set(msft.keys()) 

both = ibm_dates.intersection(msft_dates) 

for d in sorted(both): 
    print d, ibm[d], msft[d] 

Answer 2

Как это сделать:

def intersect(seq1, seq2): 
    if seq1[0] == seq2[0]: # compare the date columns 
     return (seq1[4], seq2[4]) # return 2-tuple with the cls values