deleting element from python dictionary

Question

I need the most efficient way to delete few items from the dictionary, RIght now, Im using for stmt as given below..Thinking the same thing should be accomplished in few lines.

for eachitem in dicta:
                del eachitem['NAME']
                del eachitem['STATE']
                del eachitem['COUNTRY']
                del eachitem['REGION']
                del eachitem['LNAME']

dicta = [{'name','Bob','STATE':'VA','COUNTRY':'US','REGION':'MIDWEST','LNAME':'Brian',Salary:6000}]

I want only the salary item in the dictionary once its deleted. Any inputs are appreciated.

Do you want only the salary item in the dictionary, or do you only want the salary? Just selecting that makes it a lot simpler. — Chris Morgan
– Chris Morgan, Commented Feb 21, 2013 at 18:22
I have not given the full items in the dictionary , it has totally 20 elements and I need to delete only 5 elements...#Mgilson approach is a perfect for me ..thanks — user1050619
– user1050619, Commented Feb 21, 2013 at 18:25
FWIW , efficiency with stuff like this is largely negligible. readability is often the best approach. — Jonathan Vanasco
– Jonathan Vanasco, Commented Feb 21, 2013 at 18:29

Abhijit · Accepted Answer · 2013-02-21 18:12:21Z

6

If your example data is what you are dealing with, instead of deleting the elements, just recreate your dict with that lone key

dicta = [{'Salary':e['Salary']} for e in dicta]

or to me, it makes more sense, to just create a list instead of list of dicts

dicta = [e['Salary'] for e in dicta]

but depends on what you are planning to achieve

answered Feb 21, 2013 at 18:12

Abhijit

64k20 gold badges143 silver badges209 bronze badges

Sign up to request clarification or add additional context in comments.

3 Comments

David Marx Over a year ago

Confirmed: this is the fastest method thus posted. See my post.

mgilson Over a year ago

Yeah, when I posted, there wasn't the statement that OP only wanted to keep 1 key/value pair in the dict. I've updated with a solution that uses dict.clear ... Don't know how it compares though.

David Marx Over a year ago

Oh well if all he wants is one item from the dict then clearly it would be faster to just get that one item than to go through deleting all the others.

mgilson · Accepted Answer · 2013-02-21 18:24:16Z

5

I suppose you could use:

for eachitem in dicta:
    for k in ['NAME','STATE','COUNTRY','REGION','LNAME']:
        del eachitem[k]

Or, if you only want 1 key:

for eachitem in dicta:
    salary = eachitem['SALARY']
    eachitem.clear()
    eachitem['SALARY'] = salary

This does everything in place which I assume you want -- Otherwise, you can do it out of place simply by:

eachitem = {'SALARY':eachitem['SALARY']}

edited Feb 21, 2013 at 18:24

answered Feb 21, 2013 at 18:06

mgilson

312k70 gold badges656 silver badges722 bronze badges

Comments

David Marx · Accepted Answer · 2013-02-21 18:42:24Z

EDIT: I had said dict comprehension was fastest: I was wrong. I had accidentally run %timeit test3 instead of %timeit test3(). See results below.

def test1():
    dicta = [{'NAME':'Bob','STATE':'VA','COUNTRY':'US','REGION':'MIDWEST','LNAME':'Brian','Salary':6000}]
    remove = ['NAME','STATE','COUNTRY','REGION','LNAME']
    for d in dicta:
        for r in remove:
            d.pop(r)

def test2():
    dicta = [{'NAME':'Bob','STATE':'VA','COUNTRY':'US','REGION':'MIDWEST','LNAME':'Brian','Salary':6000}]
    remove = ['NAME','STATE','COUNTRY','REGION','LNAME']
    for d in dicta:
        for r in remove:
            del d[r]

def test3():
    dicta = [{'NAME':'Bob','STATE':'VA','COUNTRY':'US','REGION':'MIDWEST','LNAME':'Brian','Salary':6000}]
    remove = ['NAME','STATE','COUNTRY','REGION','LNAME']    
    dicta = [{k:v for k,v in d.iteritems() if k not in remove } for d in dicta]

# this is really what OP was looking for, the other 3 tests are more generalized.    
def test4():    
    dicta = [{'NAME':'Bob','STATE':'VA','COUNTRY':'US','REGION':'MIDWEST','LNAME':'Brian','Salary':6000}]
    remove = ['NAME','STATE','COUNTRY','REGION','LNAME']    
    dicta = [e['Salary'] for e in dicta]

%timeit test1()
# 100000 loops, best of 3: 2.32 us per loop    
%timeit test2()
# 1000000 loops, best of 3: 1.68 us per loop    
%timeit test3()
# 100000 loops, best of 3: 3.23 us per loop
%timeit test4()
# 1000000 loops, best of 3: 1.46 us per loop

+1: Cannot resist to vote, btw interesting finding d.pop is slower than del, may be because of function call overhead
Updated post. I messed up my call to timeit, apparently the conditional dict comprehension is not faster than just using del
This gives me a invalid syntax error-dicta = [{k:v for k,v in d.iteritems() if k not in remove } for d in dicta]
that dict comprehension syntax is only supported for python 2.7+ stackoverflow.com/a/1747827/819544 . Try this: dicta = [dict((k,v) for k,v in d.iteritems() if k not in remove ) for d in dicta]

Collectives™ on Stack Overflow

deleting element from python dictionary

3 Answers 3

3 Comments

Comments

4 Comments

Linked

Hot Network Questions

Collectives™ on Stack Overflow

3 Answers 3

3 Comments

Comments

4 Comments

Linked

Related